[英]Unsuccessful in Appending Numpy Arrays
我试图遍历CSV文件并为文件中的每一行创建一个numpy数组,其中第一列表示x坐标,第二列表示y坐标。 然后我试图将每个数组附加到主数组并返回它。
import numpy as np
thedoc = open("data.csv")
headers = thedoc.readline()
def generatingArray(thedoc):
masterArray = np.array([])
for numbers in thedoc:
editDocument = numbers.strip().split(",")
x = editDocument[0]
y = editDocument[1]
createdArray = np.array((x, y))
masterArray = np.append([createdArray])
return masterArray
print(generatingArray(thedoc))
我希望看到一个包含所有CSV信息的数组。 相反,我收到一个错误:“append()缺少1个必需的位置参数:'values'非常感谢我的错误在哪里以及如何解决它的任何帮助!
Numpy数组并没有像python列表那样神奇地增长。 在添加所有内容之前,需要在“masterArray = np.array([])”函数调用中为数组分配空间。
最好的答案是使用genfromtxt( https://docs.scipy.org/doc/numpy-1.10.1/user/basics.io.genfromtxt.html )直接导入numpy数组,但......
如果你知道你正在阅读的行数,或者你可以使用这样的东西得到它。
file_length = len(open("data.csv").readlines())
然后你可以预先分配numpy数组来做这样的事情:
masterArray = np.empty((file_length, 2))
for i, numbers in enumerate(thedoc):
editDocument = numbers.strip().split(",")
x = editDocument[0]
y = editDocument[1]
masterArray[i] = [x, y]
我会推荐第一种方法但是如果你很懒,那么你总是可以构建一个python列表然后创建一个numpy数组。
masterArray = []
for numbers in thedoc:
editDocument = numbers.strip().split(",")
x = editDocument[0]
y = editDocument[1]
createdArray = [x, y]
masterArray.append(createdArray)
return np.array(masterArray)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.