Question

我希望我的代码遍历列，为每列创建一个绘图。我有以下代码为一列绘制绘图，但我不知道如何使用此代码循环遍历其他列并生成所有列的绘图。有什么想法吗？

这是我的代码：

import seaborn as sns

sns.set()
fig, ax = plt.subplots()
sns.set(style="ticks")
sns.boxplot(x='Location', y='Height [cm]', data=df) 
sns.despine(offset=10, trim=True) 
fig.set_size_inches(22,14)
plt.savefig('Height.pdf', bbox_inches='tight')

这就是我的数据：

Location            Height          Length       Width    
A                    150             95           18
A                    148             122          25
A                    162             127          16
B                    155             146          32
B                    148             112          21
B                    154             108          30
C                    160             127          22
C                    148             138          36
C                    159             142          28

Answer 1

只需将代码放入循环中并每次都更改列名和绘图名称（通过快速测试，它对我有用，并且我在工作目录中保存了3个PDF）：

import matplotlib.pyplot as plt
import seaborn as sns

for column in df.columns[1:]:  # Loop over all columns except 'Location'
    sns.set()
    fig, ax = plt.subplots()
    sns.set(style="ticks")
    sns.boxplot(x='Location', y=column, data=df)  # column is chosen here
    sns.despine(offset=10, trim=True) 
    fig.set_size_inches(22,14)
    plt.savefig('{}.pdf'.format(column), bbox_inches='tight')  # filename chosen here

Answer 2

仅绘制特定的列列表列出要绘制的列然后遍历列表在这里，我正在绘制回归图，长度与高度和宽度与高度，循环中。

import matplotlib.pyplot as plt

import seaborn as sns
lst = [ "Length"  , "Width" ]
for i in lst:
    sns.lmplot(x = 'Height', y = i , data = df ) # df is your dataframe

迭代列以在python中生成单独的图

2 个答案: