如何为熊猫数据框生成连续名称列表

【问题标题】：How can I generate a list of continuous names for a pandas dataframe如何为熊猫数据框生成连续名称列表
【发布时间】：2022-09-23 01:59:32
【问题描述】：

我想生成一个如下列表：

1a 2a 3a 4a 5a

用于熊猫数据框，到目前为止，我尝试过使用：

df.set_index, & df.index.astype(str)

但这会产生两个数据框，每个数据框都有一个名称例如

 1a 2a 3a 4a

1
2
3
4
  1  2  3  4
1b
2b
3b
4b

与

  1a 2a 3a 4a
1b
2b
3b
4b

转置？ df2 = df1.T
[f\'{i+1}a\' for i in range(5)] 生成列表

标签： python pandas dataframe columnname

【解决方案1】：

df = pd.DataFrame({1: [5, 6, 7, 8, 9],
                   2: [5, 6, 7, 8, 9],
                   3: [5, 6, 7, 8, 9],
                   4: [5, 6, 7, 8, 9],
                   5: [1, 2, 3, 4, 5]})

print(df)

   1  2  3  4  5
0  5  5  5  5  1
1  6  6  6  6  2
2  7  7  7  7  3
3  8  8  8  8  4
4  9  9  9  9  5

如果您的数据框列标题是上面的 1 到 5 号，您可以使用 rename 方法：

df.rename(columns=lambda x: str(x)+'a', inplace=True)

如果没有，您可以用set_axis 方法替换整个列标题。

df.set_axis([f'{i+1}a' for i in range(5)], axis=1, inplace=True)

两种方法都会给出输出

   1a  2a  3a  4a  5a
0   5   5   5   5   1
1   6   6   6   6   2
2   7   7   7   7   3
3   8   8   8   8   4
4   9   9   9   9   5

【讨论】：