转换为 Pandas 数据框

Question

我将 csv 转换为 pandas 数据框，但是当我运行代码时，它找不到列。在分离函数中进行转换的目的是因为我希望能够对我的函数进行单元测试。

import pandas as pd

def main():
    inputDf = pd.read_csv("android-games.csv", header=0, delim_whitespace=True)

    df = inputDf.transform(model)

    print(df)


def model(df):
    df\
        .transform(add_col)\
        .transform(test_col)

def add_col(df):
    df["price_1"] = df["price"] + 1

def test_col(df):
    df["rank_1"] = df["rank"] + 1


if __name__ == '__main__':
    main()

这是我的错误，我猜我的代码结构不好：

文件 "/Library/Frameworks/Python.framework/Versions/3.8/lib/python3.8/site-packages/pandas/core/indexes/range.py", 第 358 行，在 get_loc 中 raise KeyError(key) KeyError: 'price'

Answer 1

我不清楚您为什么要使用转换，因为我认为您可以通过以下更简单的方式获得结果（除非我误解了您想要实现的目标）：

df = pd.read_csv("android-games.csv", header=0, delim_whitespace=True)
df["price_1"] = df["price"] + 1
df["rank_1"] = df["rank"] + 1
print(df)