Моя цель — создать функцию, которая принимает на вход CSV-файл. Из этого файла CSV я хочу, чтобы он создал массив, в котором данные в первом столбце — это координата x, а данные во втором столбце — это координата y. Я хочу, чтобы для каждой строки файла данных добавлялись данные в массив.
Код: Выделить всё
import numpy as np
doc = open("d.csv")
headers = doc.readline()
def generateArray(doc):
for theData in doc:
editDocument = theData.strip().split(",")
x = splitDocument[0]
y = splitDocument[1]
createArray = np.array((x, y))
return createArray
print(generateArray(doc))
Когда я возвращаю его, он просто возвращает последнюю строку файла CSV, хотя на самом деле я хочу, чтобы все созданные массивы возвращались. Функции, которые я использовал, например .append(), выдают ошибку, говорящую, что «добавление нельзя использовать в 0-мерном массиве». Любые предложения о том, как я могу отредактировать свой код, чтобы получить возвращаемые массивы.
Подробнее здесь:
https://stackoverflow.com/questions/555 ... turning-it