python 列出面板数据所有变量名的示例代码

import pandas as pd  
# 假设我们有一个面板数据DataFrame，其中有两层索引：'year' 和 'id'  
# 并且有多个变量（列）：'var1', 'var2', 'var3'  
# 创建示例数据  
data = {  
    'year': [2020, 2020, 2021, 2021, 2020, 2020, 2021, 2021],  
    'id': [1, 2, 1, 2, 3, 4, 3, 4],  
    'var1': [10, 20, 30, 40, 50, 60, 70, 80],  
    'var2': [100, 200, 300, 400, 500, 600, 700, 800],  
    'var3': [1000, 2000, 3000, 4000, 5000, 6000, 7000, 8000]  
}  
# 将数据转换为DataFrame，并设置'year'和'id'为索引  
df = pd.DataFrame(data).set_index(['year', 'id'])  
# 检查DataFrame的结构  
print(df)  
# 列出所有变量名（列名）  
variable_names = df.columns.tolist()  
print("所有变量名（列名）:")  
for var_name in variable_names:  
    print(var_name)

import pandas as pd  
# 假设 df 是你的 DataFrame  
# ...  
# 直接使用 columns 属性  
variable_names = df.columns  
print("所有变量名（列名）:")  
for var_name in variable_names:  
    print(var_name)

# 使用列表推导式  
variable_names = [col for col in df.columns]  
print("所有变量名（列名）:")  
for var_name in variable_names:  
    print(var_name)

import numpy as np  
# 转换为 NumPy 数组  
variable_names_np = np.array(df.columns)  
# 但通常你会直接迭代或转换为列表  
variable_names_list = variable_names_np.tolist()  
print("所有变量名（列名）:")  
for var_name in variable_names_list:  
    print(var_name)