gpt4 book ai didi

python - 如何组合一个 numpy 数组和一个文本列并导出到 csv

转载 作者:太空宇宙 更新时间:2023-11-03 13:44:16 24 4
gpt4 key购买 nike

我想组合一个 numpy 数组和一个列,其中我是一个字符串作为标识符,用于导出到一个 csv 文件,然后我可以将该文件导入到 excel 中。

例如:

a=np.random.rand(6,4)
b=[]
for i in range(6):
b.append('test')

所以现在我想将 b 附加到 a 的最后一列

一旦完成,我想使用 np.savetxt(或类似的)将数组写入文件。

非常感谢任何帮助。

最佳答案

import numpy as np

a = np.random.rand(6,4)
b = ['test']*6

c = np.column_stack([a,b])
np.savetxt('/tmp/out', c, delimiter=',', fmt='%s')

像这样写

0.70503807191,0.19298150889,0.962915679186,0.655430709887,test
0.586655200042,0.379720344068,0.136924270418,0.547277504174,test
0.777238053817,0.642467338742,0.709351872598,0.932239808362,test
0.386983024375,0.753005132745,0.124107902275,0.472997270033,test
0.169711196953,0.735713880779,0.280588048467,0.726851876957,test
0.20578446385,0.379406838045,0.640154333103,0.579077700263,test

/tmp/out


跟进保罗的建议,如果你有 pandas , 你可以很容易地形成一个 DataFrame 然后调用它的 to_csv方法:

import numpy as np
import pandas as pd

a = np.random.rand(6,4)
b = np.asarray(['test']*6)

df = pd.DataFrame(a)
df['b'] = b

df.to_csv('/tmp/out')

关于python - 如何组合一个 numpy 数组和一个文本列并导出到 csv,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/23687990/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com