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)