以下是一个示例代码,它遍历Pandas DataFrame的列名列表,并使用响应变量构建条形图。
import pandas as pd
import matplotlib.pyplot as plt
# 创建一个示例DataFrame
data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}
df = pd.DataFrame(data)
# 响应变量
response_variable = 'B'
# 遍历列名列表
for column in df.columns:
# 如果列名与响应变量匹配
if column == response_variable:
# 获取列数据
column_data = df[column]
# 构建条形图
plt.bar(df.index, column_data)
plt.xlabel('Index')
plt.ylabel(column)
plt.title('Bar Chart for ' + column)
plt.show()
这个示例代码首先创建一个示例DataFrame,然后定义了一个响应变量(在这里是列名为'B'的列)。然后,使用for循环遍历DataFrame的列名列表。如果列名与响应变量匹配,就获取该列的数据,并使用matplotlib库构建条形图。最后,显示每个列名对应的条形图。