获取值太多而无法解压缩的错误(expected2)

2024-04-16 05:50:55 发布

您现在位置:Python中文网/ 问答频道 /正文

我想从我的函数中返回两个不同的值。我如何重写代码以在两个不同的变量中返回这两个值

所以我得到了这个错误:


Traceback (most recent call last):
  File "features.py", line 300, in <module>
    fea1, fea2 = build_feature_matrix_S(sentences)
ValueError: too many values to unpack (expected 2)

这是密码

def build_feature_matrix_S(documents):

    feature_type = ['frequency','tfdif']
    #feature_type = feature_type.lower().strip()

    for f in feature_type:

        if feature_type == 'frequency':
            vectorizer_freq = CountVectorizer(tokenizer = tokenize_lemmatize_spacy, binary=False, min_df=5, ngram_range=(1, 2))
            feature_matrix_freq = vectorizer_freq.fit_transform(documents)
            feature_freq = pd.DataFrame(feature_matrix_freq.todense(), columns = vectorizer_freq.get_feature_names())

            return feature_freq

        else:

            vectorizer_tfidf = TfidfVectorizer(tokenizer = tokenize_lemmatize_spacy, min_df=5, ngram_range=(1, 2))
            feature_matrix_tfidf = vectorizer_tfidf.fit_transform(documents)
            feature_tfidf = pd.DataFrame(feature_matrix_tfidf.todense(), columns = vectorizer_tfidf.get_feature_names())

            return feature_tfidf

fea1, fea2 = build_feature_matrix_S(sentences)

print(fea1, fea2)


Tags: inbuildtypesentencesmatrixdocumentsfeaturetokenizer
2条回答

在if、else语句中,只返回1个变量。您可以从如下函数中返回两个变量:

return var1, var2

因此,您的函数可以写入如下内容:

def build_feature_matrix_S(documents):

    # part 1:
    vectorizer_freq = CountVectorizer(tokenizer = tokenize_lemmatize_spacy, binary=False, min_df=5, ngram_range=(1, 2))
    feature_matrix_freq = vectorizer_freq.fit_transform(documents)
    feature_freq = pd.DataFrame(feature_matrix_freq.todense(), columns = vectorizer_freq.get_feature_names())

    # part 2:
    vectorizer_tfidf = TfidfVectorizer(tokenizer = tokenize_lemmatize_spacy, min_df=5, ngram_range=(1, 2))
    feature_matrix_tfidf = vectorizer_tfidf.fit_transform(documents)
    feature_tfidf = pd.DataFrame(feature_matrix_tfidf.todense(), columns = vectorizer_tfidf.get_feature_names())

    return feature_freq, feature_tfidf

在这方面,

fea1, fea2 = build_feature_matrix_S(sentences)

python希望返回两个变量,一个元组,函数build_feature_matrix_S()中的两个返回语句都没有返回,返回的变量超过2个

    feature_freq = pd.DataFrame(feature_matrix_freq.todense(), columns = vectorizer_freq.get_feature_names())

    return feature_freq

            feature_tfidf = pd.DataFrame(feature_matrix_tfidf.todense(), columns = vectorizer_tfidf.get_feature_names())

            return feature_tfidf

这就是错误

相关问题 更多 >