Python 重命名 Pandas DataFrame 列

Question

import pandas as pd
import numpy as np
datain = np.loadtxt(datafile)
df = pd.DataFrame(data = datain, columns = ["t","p","x","y","z"])
avg = df.groupby(["t"], sort=False)["p"].mean().rename(columns={1:"mean"})

这不起作用，它告诉我 TypeError: rename() 获得了意外的关键字参数“columns”。如果我这样做也行不通，

avg.rename(columns = {1:"mean"}, inplace=True)

我不明白为什么，所有文档都告诉我我的列调用是正确的。我只想将我的“mean”调用创建的空白列重命名为具有字符串索引。有人知道为什么或如何解决这个问题吗？我见过的所有例子都遵循这种格式。谢谢。

Answer 1

IIUC 你可以做到这一点

import pandas as pd
df = pd.DataFrame({"a":np.arange(10),
                   "b":np.random.choice(["A","B"],10)})

avg = df.groupby("b", sort=False)["a"].mean()\
        .reset_index(name="mean")

或

avg = df.groupby("b", sort=False)["a"].mean().reset_index()\
        .rename(columns={"a":"mean"})

或

avg = df.groupby("b", sort=False, as_index=False)["a"].mean()\
        .reset_index()\
        .rename(columns={"a":"mean"})