如何在将未堆叠的 Pandas 数据框传递给绘图函数之前将其重塑为“长”形式答案

【问题标题】：How to reshape an unstacked Pandas data frame to "long" form before passing it to a plotting function如何在将未堆叠的 Pandas 数据框传递给绘图函数之前将其重塑为“长”形式
【发布时间】：2020-10-04 08:46:14
【问题描述】：

我正在尝试使用 Plotly px.bar() 函数制作一个显示比率的简单条形图。

我有以下数据集：

test_df = pd.DataFrame({'Manufacturer':['Ford', 'Ford', 'Mercedes', 'BMW', 'Ford', 'Mercedes', 'BMW', 'Ford', 'Mercedes', 'BMW', 'Ford', 'Mercedes', 'BMW', 'Ford', 'Mercedes', 'BMW', 'Ford', 'Mercedes', 'BMW'],
                          'Metric':['Orders', 'Orders', 'Orders', 'Orders', 'Orders', 'Orders', 'Orders', 'Sales', 'Sales', 'Sales', 'Sales', 'Sales', 'Sales', 'Warranty', 'Warranty', 'Warranty', 'Warranty', 'Warranty', 'Warranty'],
                          'Sector':['Germany', 'Germany', 'Germany', 'Germany', 'USA', 'USA', 'USA', 'Germany', 'Germany', 'Germany', 'USA', 'USA', 'USA', 'Germany', 'Germany', 'Germany', 'USA', 'USA', 'USA'],
                          'Value':[45000, 70000, 90000, 65000, 40000, 65000, 63000, 2700, 4400, 3400, 3000, 4700, 5700, 1500, 2000, 2500, 1300, 2000, 2450],
                          'City': ['Frankfurt', 'Bremen', 'Berlin', 'Hamburg', 'New York', 'Chicago', 'Los Angeles', 'Dresden', 'Munich', 'Cologne', 'Miami', 'Atlanta', 'Phoenix', 'Nuremberg', 'Dusseldorf', 'Leipzig', 'Houston', 'San Diego', 'San Francisco']
                   })

我重置了索引，创建了一个数据透视表，如下：：

temp_table = test_df.reset_index().pivot_table(values = 'Value', index = ['Manufacturer', 'Metric', 'Sector'], aggfunc='sum')

然后，我创建了两个新的数据框：

s1 = temp_table.set_index(['Manufacturer','Sector']).query("Metric=='Orders'").Value
s2 = temp_table.set_index(['Manufacturer','Sector']).query("Metric=='Sales'").Value

然后，我解开这些数据帧：

s1.div(s2).unstack()

这给了我：

Sector             Germany      USA
Manufacturer
---
BMW               19.117647     11.052632
Ford              42.592593     13.333333
Mercedes          20.454545     13.829787

我希望能够使用上面的数据制作条形图，Manufacturer 在 x 轴上，并由 Sector 着色，如下所示：

为此，我认为我需要将数据采用以下长格式：

Manufacturer       Sector        Ratio
BMW                Germany       19.117647
Ford               Germany       42.592593
Mercedes           Germany       20.454545
BMW                USA           11.052632
Ford               USA           13.333333
Mercedes           USA           13.829787

问题：我将如何重塑上面未堆叠的数据，以便能够将其传递给 Plotly px.bar() 函数，该函数需要以下 x 轴和 y 轴参数：

x (str or int or Series or array-like) – data_frame 中列的名称，或者 pandas Series 或 array_like 对象。此列或 array_like 中的值用于在笛卡尔坐标中沿 x 轴定位标记。 x 或 y 可以选择是列引用列表或 array_likes，在这种情况下，数据将被视为“宽”而不是“长”。

提前致谢！

【问题讨论】：

你可以试试test_df.groupby(['Manufacturer', 'Sector'])['Value'].sum().unstack('Sector').plot.bar()

标签： pandas plotly plotly-python

【解决方案1】：

这应该会给你上面的条形图。

test_df.groupby(['Manufacturer', 'Sector'])['Value'].sum().unstack('Sector').plot.bar()

【讨论】：

我可以使用 .plot.bar() 函数...但是，我想使用 Plotly 函数，这需要重塑。

【解决方案2】：

不要做unstack

df_out=s1.div(s2).reset_index()

【讨论】：