как выполнить подзаговор в цикле для морских графиков - PullRequest
Новая
       15

как выполнить подзаговор в цикле для морских графиков

1 голос
/ 04 октября 2019

У меня есть код, который генерирует четыре вспомогательных графика, но я хочу сгенерировать эти диаграммы с помощью циклов. В настоящее время я слежу за этим фрагментом кода, чтобы сгенерировать диаграмму. Код: 

plt.figure(figsize=(20, 12))
plt.subplot(221)
sns.barplot(x = 'Category', y = 'POG_Added', data = df)
xticks(rotation = 90)
plt.xticks(size = 11)
plt.yticks(size = 11)
plt.xlabel("Category",size = 13)
plt.ylabel("POG_Added",size = 13)

plt.subplot(222)
sns.barplot(x = 'Category', y = 'Live_POG', data = df)
xticks(rotation = 90)
plt.xticks(size = 11)
plt.yticks(size = 11)
plt.xlabel("Category",size = 13)
plt.ylabel("Live_POG",size = 13)

plt.subplot(223)
sns.lineplot(x = 'Category', y = 'D01_CVR', data = df)
#sns.barplot(x = 'Category', y = 'D2-08-Visits', data = df,label='D2-08_Visits')
xticks(rotation = 90)
plt.xticks(size = 11)
plt.yticks(size = 11)
plt.xlabel("Category",size = 13)
plt.ylabel("D01_CVR",size = 13)

plt.subplot(224)

plt.xticks(rotation='vertical')
ax = sns.barplot(x='Category',y='D2-08-Units',data=df)
ax2 = ax.twinx()
ax2.plot(ax.get_xticks(), df["D01_CVR"], alpha = .75, color = 'r')

plt.subplots_adjust(hspace=0.55,wspace=0.55)
plt.show()

enter image description here

Ответы [ 2 ]

0 голосов
/ 04 октября 2019

Рассмотрите возможность ужесточения повторяющегося кода с помощью:

  • Установите неизменную эстетику, такую ​​как все размеры шрифтов x-ticks и y-ticks, за один вызов с помощью вызовов plt.rc.
  • Построить plt.subplots() и использовать его массив объектов Axes.
  • Использовать ax аргумент Seaborn's barplot и lineplot для петли над массивом осей.

Хотя не полностью СУХОЙ, учитывая два специальных графика, ниже приведена настройка: 

# AXES AND TICKS FONT SIZES
plt.rc('xtick', labelsize=11)
plt.rc('ytick', labelsize=11)
plt.rc('axes', labelsize=13)

# FIGURE AND SUBPLOTS SETUP
fig, axes = plt.subplots(nrows=2, ncols=2, figsize=(20, 12))

# BAR PLOTS (FIRST ROW)
for i, col in enumerate(['POG_Added', 'Live_POG']):
    sns.barplot(x='Category', y=col, data=df, ax=axes[0,i])
    axes[0,i].tick_params(axis='x', labelrotation=90)

# LINE PLOT 
sns.lineplot(x='Category', y='D01_CVR', data=df, ax=axes[1,0])
axes[1,0].tick_params(axis='x', labelrotation=90)

# BAR + LINE DUAL PLOT
sns.barplot(x='Category', y='D2-08-Units', data=df, ax=axes[1,1])
ax2 = axes[1,1].twinx()
ax2.plot(axes[1,1].get_xticks(), df["D01_CVR"], alpha = .75, color = 'r')
axes[1,1].tick_params(axis='x', labelrotation=90)
0 голосов
/ 04 октября 2019

Вот как я делаю такие вещи: 

import numpy as np
import matplotlib.pyplot as plt

data = [np.random.random((10, 10)) for _ in range(6)]

fig, axs = plt.subplots(ncols=3, nrows=2, figsize=(9, 6))
for ax, dat in zip(axs.ravel(), data):
    ax.imshow(dat)

Это дает:

matplotlib output

Идея состоит в том, что plt.subplots() создает массив Axes объектов, так что вы можете зацикливаться на нем и создавать свои графики в цикле. В этом случае мне нужно ndarray.ravel(), потому что axs - это двумерный массив.

...