首页 > 解决方案 > 将 csv 文件转换为数组 uisng pandas

问题描述

我有一个 csv 文件,如下所示

年,增长
1, 10
2, 12
3, 13
4, 15

我想将 csv 文件的内容转换为一个数组,比如 my_array,这样我就可以访问增长,如下所示

my_array[1] 应该返回 10

my_array[4] 应该返回 15 等

如何使用 pandas 将 csv 转换为这样的数组?

标签: python-3.x

解决方案


>>> import pandas
>>> df = pandas.read_csv('test.csv')
>>> df
   Year  Growth
0      1      10
1      2      12
2      3      13
3      4      15
>>> arr = df['Growth'].values.tolist()
>>> arr[0]
10

是的,上面的代码将帮助你为你想要的任何列做到这一点。


推荐阅读