python - 使用 Matplotlib 和 csv2rec 设置数据类型

我正在学习 Matplotlib，并尝试手动实现一个简单的线性回归。
但是，我在使用 csv2rec 后导入然后处理我的数据时遇到了问题。

data= matplotlib.mlab.csv2rec('KC_Filtered01.csv',delimiter=',')

x = data['list_price']
y = data['square_feet']

sumx = x.sum()
sumy = y.sum()

sumxSQ = sum([sq**2 for sq in x])
sumySQ = sum([sq**2 for sq in y])

我正在阅读房价列表，并试图获得平方和。但是，当 csv2rec 从文件中读取价格时，它会将这些值存储为 int32。由于房价的平方和大于32位整数，所以溢出。但是，我没有看到更改 csv2rec 读取文件时分配的数据类型的方法。读入或分配数组时如何更改数据类型？

最佳答案

x = data['list_price'].astype('int64')

y 也是如此。

并且:csv2rec 有一个 converterd 参数:http://matplotlib.sourceforge.net/api/mlab_api.html#matplotlib.mlab.csv2rec

关于python - 使用 Matplotlib 和 csv2rec 设置数据类型，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/7366302/