重命名变量并使用向量中的字符串存储“选择”该变量

Question

我是一名 R 用户，正在尝试学习 python。

为了进行一些分析，我曾经像这样重命名数据框变量

library(dplyr) 

variable = "c"
df = data.frame(a=c(8,5,7,8), b=c(9,6,6,8), c=c(0,7,8,9))

> df
  a b c
1 8 9 0
2 5 6 7
3 7 6 8
4 8 8 9

out = df %>% rename(variable := !!variable)

> out
 
 a b variable
1 8 9        0
2 5 6        7
3 7 6        8
4 8 8        9

我找不到在 python 中做同样事情的方法。有什么想法可以帮助我吗？

import pandas as pd

df = pd.DataFrame(
    {
        "a": [8,5,7,8],
        "b": [9,6,6,8],
        "c": [0,7,8,9],
       
    }
)

Answer 1

您可以使用

rename

和字典做同样的事情：

variable = 'c'
df.rename(columns={variable: 'variable'})

输出：

   a  b  variable
0  8  9         0
1  5  6         7
2  7  6         8
3  8  8         9