Python，ValueError：x 和 y 必须具有相同的第一维问题

Question

我试图在 CSV 文件和 plot 中读取，但是当我尝试和 plot y 变量时，它说该变量只有两个元素，当后面应该遵循代码时：

import numpy as np
import csv

x = np.array([1,2,3,4,5,6])
y1=[]
y2=[]

with open('file.csv', 'r') as csvfile:
plots= csv.reader(csvfile, delimiter=',')
for row in plots:
    y1.append(int(row[1]))
print(len(y1))


with open('file.csv', 'r') as csvfile:
plots= csv.reader(csvfile, delimiter=',')
for row in plots:
    y2.append(int(row[2]))
print(len(y2))

这会产生 2,2，这是问题所在，它应该产生 6,6，因为两者都有 6 个元素？ CSV 文件设置为以下格式：Dishwasher,1,1,1,1,1,1

洗衣机,1,1,0,1,1,0

Answer 1

它产生 2,2 因为plots中有 2 个row 。 本质上，您将第二列存储在 y1 中，将第三列存储在 y2 中。

如果这是您想要的，那么您的代码就可以了，如果您想测量行长，则可以使用以下代码：

import csv

with open('file.csv', 'r') as csvfile:
    plots = csv.reader(csvfile, delimiter=',')
    rows = [row for row in plots]
    print(len(rows[0]) - 1)  # -1 to exclude title
    print(len(rows[1]) - 1)