将数组元素存入.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)