如何在CSV文件中的特定行中写入数据

Question

目前我有一个csv文件，其中最后一列（我们可以称为label）为空。这是一张数据图片：

我通过分类器得到预测：

[0.  1.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  1.  1.  1.  0.  0.  0.]

因此我的问题是如何将这些预测写入最后一列（label）？

Answer 1

最简单的解决方案之一可能是使用pandas读取数据。

my_dataframe = pandas.read_csv("filename")

然后假设您的标签列是某种列表/数组甚至是pandas系列，您只需输入

即可。

my_dataframe["label"] = my_label_list

随后您使用

回写您的pandas数据帧

my_dataframe.to_csv("filename_out")

注意：我省略了用于读取和编写csv的其他参数参数，您可以谷歌并阅读文档并根据需要添加它们。

Answer 2

您应该从csv文件中读取数据而不使用最后一列

df = pd.DataFrame(data, columns = ['domain_name_length', 'domain_name_enthropy', 'digit_num', 'alpha_num', 'delimiter_num', 'special_num','max_conitous_digit_length','max_conitous_alpha_length','max_length_between_delimeter'])

不只是添加你的专栏

df['labels'] = labels