Pandas: 다중 행 서브플롯을 플롯할 수 없음

에 만든 2014년 08월 26일 · 3코멘트 · 출처: pandas-dev/pandas

이 코드는 3개의 서브플롯으로 구성된 단일 행을 생성하는 데 잘 작동합니다.

g = np.random.choice([1,2,3], 10)
s = np.random.normal(size=10)
s2 =np.random.normal(size=10)
df = pd.DataFrame([g, s, s2]).T
df.columns = ['key', 's1', 's2']
gb = df.groupby('key')

fig, axes = plt.subplots(nrows=1, ncols=3, figsize=(12, 4))
i = 0
for key, df2 in gb:
    df2.plot(ax=axes[i], x='s1', y='s2', title=key)
    i = i + 1

그러나 두 번째 행(nrows=2)을 추가하려고 하면 속성 오류가 발생합니다.

AttributeError: 'numpy.ndarray' 개체에 'get_figure' 속성이 없습니다.

fig, axes = plt.subplots(nrows=2, ncols=3)
fig.tight_layout() # Or equivalently,  "plt.tight_layout()"
i = 1
for key, df2 in gb:
    df2.plot(ax=axes[i])
    i = i + 1

출처

ericdf

👍1

가장 유용한 댓글

두 번째 행으로 둘러싸고 싶다면 axes[i // 3][i % 3] 와 같이 해야 합니다.
(예를 6개의 그룹으로 확장했습니다)

In [67]: df
Out[67]: 
    key        s1        s2
0     3 -1.452043 -0.119374
1     1  0.603860 -1.635034
2     3  0.964165 -0.043124
3     2  0.459628 -0.538155
4     3  0.398761 -0.195261
5     1  0.085750 -0.116766
6     2 -0.397419 -0.140660
7     3 -0.053209  1.547755
8     1 -0.634555 -0.509077
9     3  0.138808  0.608165
10    6 -1.452043 -0.119374
11    4  0.603860 -1.635034
12    6  0.964165 -0.043124
13    5  0.459628 -0.538155
14    6  0.398761 -0.195261
15    4  0.085750 -0.116766
16    5 -0.397419 -0.140660
17    6 -0.053209  1.547755
18    4 -0.634555 -0.509077
19    6  0.138808  0.608165

In [63]: for i, (key, df2) in enumerate(gb):
    df2.plot(ax=axes[i // 3][i % 3])

TomAugspurger 에 2014년 08월 26일

👍4 ❤2

모든 3 댓글

axes 가 이제 matplotlib 축의 2차원 배열이기 때문입니다.

In [35]: fig, axes = plt.subplots(nrows=2, ncols=3)

In [36]: axes
Out[36]: 
array([[<matplotlib.axes._subplots.AxesSubplot object at 0x10c094ac8>,
        <matplotlib.axes._subplots.AxesSubplot object at 0x10ee40b70>,
        <matplotlib.axes._subplots.AxesSubplot object at 0x10c0ac240>],
       [<matplotlib.axes._subplots.AxesSubplot object at 0x10edf8a90>,
        <matplotlib.axes._subplots.AxesSubplot object at 0x10ec27630>,
        <matplotlib.axes._subplots.AxesSubplot object at 0x10edc9128>]], dtype=object)

In [37]: axes.shape
Out[37]: (2, 3)

다음과 같은 것을 시도하십시오.

In [38]: for i, (key, df2) in enumerate(gb):
    df2.plot(ax=axes[0][i])

TomAugspurger 에 2014년 08월 26일

👍2

두 번째 행으로 둘러싸고 싶다면 axes[i // 3][i % 3] 와 같이 해야 합니다.
(예를 6개의 그룹으로 확장했습니다)

In [67]: df
Out[67]: 
    key        s1        s2
0     3 -1.452043 -0.119374
1     1  0.603860 -1.635034
2     3  0.964165 -0.043124
3     2  0.459628 -0.538155
4     3  0.398761 -0.195261
5     1  0.085750 -0.116766
6     2 -0.397419 -0.140660
7     3 -0.053209  1.547755
8     1 -0.634555 -0.509077
9     3  0.138808  0.608165
10    6 -1.452043 -0.119374
11    4  0.603860 -1.635034
12    6  0.964165 -0.043124
13    5  0.459628 -0.538155
14    6  0.398761 -0.195261
15    4  0.085750 -0.116766
16    5 -0.397419 -0.140660
17    6 -0.053209  1.547755
18    4 -0.634555 -0.509077
19    6  0.138808  0.608165

In [63]: for i, (key, df2) in enumerate(gb):
    df2.plot(ax=axes[i // 3][i % 3])

TomAugspurger 에 2014년 08월 26일

👍4 ❤2

정말 고마워요, 톰. 이제 완벽하게 작동합니다. 도움이 되는 답변에 진심으로 감사드립니다.

ericdf 에 2014년 08월 27일

이 페이지가 도움이 되었나요?

0 / 5 - 0 등급

Pandas: 다중 행 서브플롯을 플롯할 수 없음

가장 유용한 댓글

모든 3 댓글

관련 문제