python 将数组元素存入.csv文件中;csv内部实现行列转换;

将数组元素存入.csv文件中

提前创建一个新的、空的文件test.csv,否则会出现出现文件已被使用的错误。

#存入csv
with open("test.csv", "w", encoding="utf-8") as f:
     writer = csv.writer(f)
     writer.writerow(test_predict)
     f.close()
#可以不用非写f

遇到的问题

在这里插入图片描述
原因在于这横代码writer.writerows(test_predict),其中writerows代表多横存入,writerow代表一横输入。
因此改为:writer.writerow(test_predict)即可

csv内部实现行列转换

提前创建一个新的、空的文件text.csv,否则会出现出现文件已被使用的错误。

import pandas as pd

df = pd.read_csv('test.csv')
data = df.values # data是数组,直接从文件读出来的数据格式是数组
index1 = list(df.keys()) # 获取原有csv文件的标题,并形成列表
data = list(map(list, zip(*data))) # map()可以单独列出列表,将数组转换成列表
data = pd.DataFrame(data, index=index1) # 将data的行列转换
data.to_csv('text.csv', header=0)

参考了
存入csv参考博客
横列转换参考博客

猜你喜欢

转载自blog.csdn.net/LZL2020LZL/article/details/126754655