从错误中引发 KeyError(key)

Question

我将遍历 taitanic 数据集。 这是代码：

import numpy as np
import pandas as pd

!wget "https://calmcode.io/datasets/titanic.csv"
dt = pd.read_csv("./titanic.csv", index_col=["PassengerId"])

def impute_cabin_values(X):
    for i in range(len(X)):
        print(X["Pclass"][i])

impute_cabin_values(dt)

然后我面临以下错误：

 raise KeyError(key) from err
KeyError: 0

为什么print(X["Pclass"][i])中的索引0不起作用？

Answer 1

这是因为您在此部分将 dataframe 索引设置为PassengerId 。

dt = pd.read_csv("./titanic.csv", index_col=["PassengerId"])

Answer 2

您正在按不存在的列PassengerId进行索引，并尝试获取不存在的列Pclass而不是pclass 。 假设这个。

import numpy as np
import pandas as pd

#!wget "https://calmcode.io/datasets/titanic.csv"
dt = pd.read_csv("./titanic.csv") #, index_col=["PassengerId"])

def impute_cabin_values(X):
    for i in range(len(X)):
        print(X["pclass"][i])

impute_cabin_values(dt)

结果：

1
3
1
3
1
3
3
2
...