我有一個很大的數據集,在第二列有句子和他們的情緒狀態。我開發了代碼來將它們讀作numpy數組。我需要的是,如果一個句子的情感是中性的,那么返回為真,否則返回假。if-else條件返回的每個結果都應寫入CSV文件。但是這里它只在CSV文件中寫入一次結果,而不是寫入for循環每次迭代返回的所有結果。以下是我目前掌握的代碼。在import csv
import numpy as np
with open('Book1.csv', encoding='utf8') as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
text=(row['text'])
emotion=(row['emotion'])
my_list=(text, emotion)
my_array = np.asarray(my_list)
make_array = (text, "neutral")
num_array = np.asarray(make_array)
if np.array_equal(my_array,num_array):
with open('test.csv', 'w') as csvfile:
fieldnames = ['result']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
writer.writerow({'result': 'True'})
else:
with open('test.csv', 'w') as csvfile:
fieldnames = ['result']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
writer.writerow({'result': 'False'})`
我可能在某個地方錯了。誰來幫幫我。在