TypeError：'AxesSubplot‘对象不支持索引

Question

问题：

使用海运创建一个散点图，显示市场资本的价格走势。以50种货币为例，市场价值最高的货币可以看到这一趋势。将地块大小分别设置为10英寸宽和2英寸高。

注意:这是使用Hackerrank进行的训练中的一个练习黑客马拉松问题，在我们的组织中进行。

import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd
import pickle


#File name: question.py
#initial try
''' 
def draw_scatterplot(df):
    plot, ax= plt.subplots(figsize=(10, 2))
    plot =sns.scatterplot(x="price", y='market_cap', data=df.nlargest(50,'market_cap'), ax=ax)
    plot.data = df.nlargest(50,'market_cap')
    return plot
'''
#Later after studied post in stackover flow 
def draw_scatterplot(df):
    plot, ax= plt.subplots(1, 1, figsize=(10, 2),squeeze=False,sharex=False)
    plot =sns.scatterplot(x="price", y='market_cap', data=df.nlargest(50,'market_cap'),ax=ax[0,0])
    plot.data = df.nlargest(50,'market_cap')
    return plot
    
def serialize_plot(plot, plot_dump_file):
    pickle.dump(plot,open(plot_dump_file,'wb'))
    
def main():
    file_name=input() # data_actual.csv
    df=pd.read_csv(file_name)
    
    plot = draw_scatterplot(df)
    
    serialize_plot(plot.axes, "plot_axes.pk")
    serialize_plot(plot.data, "plot_data.pk")
    
if __name__ == '__main__':
    main()

要测试程序，必须运行tests.py，同时执行同样的操作，下面显示的错误将出现：

    @classmethod
    def setup_class(cls):
        cls.aplot = pickle.load(open('data/actual_plots/aplot_axes.pk', 'rb'))
        cls.aplot_ax = cls.aplot[0][0]
        cls.gplot = pickle.load(open('plot_axes.pk', 'rb'))
>       cls.gplot_ax =  cls.gplot[0][0]
E       TypeError: 'AxesSubplot' object does not support indexing

tests.py:93: TypeError

'''

Answer 1

pickle.dump()似乎将2d数组降维如下：

import matplotlib.pyplot as plt
import pickle

names = ["group_a", "group_b", "group_c"]
values = [1, 10, 100]

fig, ax = plt.subplots(2, 3, figsize=(10, 2),squeeze=False,sharex=False)

ax[0][0].scatter(names, values)

pickle.dump(fig.axes, open('plot_axes.pk', 'wb'))

print(ax)
print(type(ax))

输出

[[<AxesSubplot:> <AxesSubplot:> <AxesSubplot:>]
 [<AxesSubplot:> <AxesSubplot:> <AxesSubplot:>]]
<class 'numpy.ndarray'>

import matplotlib.pyplot as plt
import pickle

gplot = pickle.load(open('plot_axes.pk', 'rb'))

gplot_ax = gplot[0]

print(gplot)
print(type(gplot))

输出：

[<AxesSubplot:>, <AxesSubplot:>, <AxesSubplot:>, <AxesSubplot:>, <AxesSubplot:>, <AxesSubplot:>]
<class 'list'>

因此，您可能需要使用cls.gplot[0]代替。