python - key 错误 : "[' petal length'] not in index"

Question

我在使用 Pandas 数据帧的索引时遇到一些问题。我想要做的是从 csv 文件加载数据，创建 Pandas 数据框，然后从该数据框中选择特定列并将其转换为数组。

KeyError: "['petal length'] not in index"

我正在使用的代码是

  import pandas as pd
  import numpy as np
  data = pd.read_csv('iris.csv')
  x = np.array(data[['sepal length', 'petal length']])

这是表格:

使用重新索引将为列创建缺失值。

感谢您的帮助!

Answer 1

您需要检查列名称:

print (data.columns)
Index(['SepalLength', 'SepalWidth', 'PetalLength', 'PetalWidth', 'Name'], dtype='object')

因此选择使用:

x = np.array(data[['SepalWidth', 'PetalWidth']])