如何过滤频率并将二元组添加到代码中？

Question

此输出给出一个条形图，其单词在x轴上，其频率在y轴上。但是，我想添加两个增强功能： 1）仅显示频率> 2的值 2）包含二元组

import pandas as pd
df = pd.DataFrame(['my big dog', 'my lazy cat'])
df
#     0
#0   my big dog
#1  my lazy cat

value_list = [row[0] for row in df.itertuples(index=False, name=None)]
value_list
#['my big dog', 'my lazy cat']

from sklearn.feature_extraction.text import CountVectorizer
cv = CountVectorizer()
x_train = cv.fit_transform(value_list)

x_train.toarray()

x_train_sum = x_train.sum(axis=0)
x_train_sum

result = []
for word, col in cv.vocabulary_.items():
 result.append((word, x_train_sum[0,col]))

word = []
frequency = []

for i in range(len(result)):
 word.append(result[i][0])
 frequency.append(result[i][1])

indices = np.arrange(len(results))
plt.bar(indices, frequency, color 'r')
plt.xticks(indices, word, rotation = 'vertical')

plt.tight_layout()
plt.show()

Answer 1

我不确定您所说的“包括二元组”是什么意思，但问题第一部分的答案在这里：

indices = [i for i in range(len(frequency)) if frequency[i] >= 2]
frequency = [frequency[i] for i in indices]
word = [word[i] for i in indices]

在创建图像之前添加这3行，将对大于2的频率进行过滤。