使用dataframe列值作为列名

import pandas as pd

# 创建示例DataFrame
data = {
    'A': [1, 2, 3],
    'B': ['col1', 'col2', 'col3'],
    'C': [4, 5, 6]
}
df = pd.DataFrame(data)
print("原始DataFrame:")
print(df)

   A    B  C
0  1  col1  4
1  2  col2  5
2  3  col3  6

# 使用列 'B' 的值作为新的列名
new_columns = df['B'].tolist()
new_df = df.rename(columns=dict(zip(df.columns[1:], new_columns)))
new_df = new_df.drop(columns=['B'])  # 删除原来的列 'B'
print("新的DataFrame:")
print(new_df)

   A  col1  col2  col3
0  1     4     5     6
1  2     4     5     6
2  3     4     5     6

# 添加后缀以避免重复列名
new_columns = [f"{col}_{i}" if i > 0 else col for i, col in enumerate(new_columns)]
new_df = df.rename(columns=dict(zip(df.columns[1:], new_columns)))
new_df = new_df.drop(columns=['B'])
print("新的DataFrame:")
print(new_df)

# 填充空值
df['B'] = df['B'].fillna('unknown')
new_columns = df['B'].tolist()
new_df = df.rename(columns=dict(zip(df.columns[1:], new_columns)))
new_df = new_df.drop(columns=['B'])
print("新的DataFrame:")
print(new_df)