如果不做判断直接使用loc[]
提取内容,会报错KeyError
,因此可以使用下面的方法:
首先构造dataframe
:
import pandas as pd
df: pd.DataFrame = pd.DataFrame([
[1, 4.2],
[70, 0.03],
[5, 0]
], columns=['one', 'two'])
生成如下矩阵:
one | two | |
---|---|---|
0 | 1 | 4.2 |
1 | 70 | 0.03 |
2 | 5 | 0 |
然后判断index是否包含给定的值:
if 1 in df.index.values:
print(df.loc[1])
else:
print('None')
注:要确保index的类型为 int 还是 str等,str(‘1’)不能等价于int(1)