gpt4 book ai didi

python - Numpy - 将第一行的 csv 作为名称立即加载到结构化数组?

转载 作者:太空宇宙 更新时间:2023-11-04 02:46:32 26 4
gpt4 key购买 nike

有没有办法避免必须在 numpy/pandas 中预定义列标题的名称来创建结构化数组,而是让 numpy/pandas 在第一行中读取作为标题名称,然后加载其余数据到结构化数组?例如。对于这些数据,

#        Date     low    open   close
# 2015-08-08 266.00 280.04 266.82
# 2015-07-08 233.00 280.04 266.82

我想调用一些像 a = pd.read_csv() 这样的函数,并且能够立即以 a['Date'] 的形式访问列,而无需预定义names = [Date, low, open, close] 并且必须将其传递到 read_csv()。这可能吗?

最佳答案

为了完整起见,一个numpy的例子:

rec_arr = np.genfromtxt("try.csv", delimiter = " ", names=True, dtype=None)
rec_arr
array([(b'2015-08-08', 266., 280.04, 266.82),
(b'2015-07-08', 233., 280.04, 266.82)],
dtype=[('Date', 'S10'), ('low', '<f8'), ('open', '<f8'), ('close', '<f8')])

然后,您可以访问如下列:

rec_arr['close']
array([ 266.82, 266.82])

像往常一样做一些数学运算:

rec_arr['close'].mean()
266.81999999999999

关于python - Numpy - 将第一行的 csv 作为名称立即加载到结构化数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44993464/

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