如果列标题中的值，如何替换值？

Question

我有如下数据框。 如果 A1 中的每列标题，我想将值替换为 1。

这是我的预期输出。

Answer 1

让您的数据框df ，并假设 2, 3, ... 列是单词。

然后使用以下代码之一。

代码 1. 使用.str.contains()

word_list = df.columns.tolist()[1:]
for word in word_list:
    df[word] = df['A1'].str.contains(word).astype(int)

代码 2. 使用sklearn.feature_extraction.text.CountVectorizer

from sklearn.feature_extraction.text import CountVectorizer
word_list = df.columns.tolist()[1:]

vectorizer = CountVectorizer(vocabulary = word_list)
X = vectorizer.fit_transform(df['A1']).toarray()
X = pd.DataFrame(X, columns = word_list)
output = pd.concat([df['A1'], X], axis = 1)