使用pandas过滤excel数据时的KeyError

Question

我正在尝试从Python中读取和过滤excel数据。我使用下面的代码：

import pandas as pd
import numpy as np
df = pd.read_excel('file.xlsx') 
df['apples'] = (pd.cut(df['apples'],bins=[-np.inf,2,5,np.inf],labels=['WOW','ok','BOB']))
print(df)

This is my excel file

但KeyError：＆＃39; apples＆＃39;发生。你有什么建议我怎么解决这个问题？

Answer 1

您是否还想修改xlsx文件？或者您只是想阅读并应用一些代码？ 在第二种情况下，您可以这样做：

df = df.drop(['apples'])

你可以输入：

inputX = df.loc[:, ['oranges', 'lemons']].as_matrix()

这取决于你想用它做什么。

Answer 2

如果您的标题包含2行，则会出现问题，因此默认情况下，DataFrame列会在第一行创建。

所以需要跳过第一行：

df = pd.read_excel('file.xlsx', skiprows=1)

或者：

df = pd.read_excel('file.xlsx', header=1)