首页 >后端开发 >Python教程 >当数据分成多列时,如何用 Pandas 绘制堆积条形图?

当数据分成多列时,如何用 Pandas 绘制堆积条形图?

Susan Sarandon
Susan Sarandon原创
2024-10-21 19:40:031049浏览

How to Plot a Stacked Bar Chart with Pandas When Data is Separated into Multiple Columns?

使用 Pandas 绘制堆积条形图

在 Python 中,我们可以使用 Pandas 和 Matplotlib 来创建堆积条形图。一个常见的挑战是构建图表的数据。

例如,考虑创建一个堆叠条形图的任务,其中数据分为多个列。给定的示例显示了一个包含网站名称和滥用/NFF 计数的电子表格。要绘制此数据:

  1. 导入库:首先导入 Pandas 和 Matplotlib。
  2. 创建数据框:创建 Pandas 数据框来自 CSV 数据。
  3. 重组数据:使用 groupby() 和 unstack() 函数将数据重组为适合条形图的格式。在示例中,数据按站点名称和滥用/NFF 进行分组,然后对计数进行堆叠。
  4. 创建条形图:使用带有 kind=' 的plot() 函数bar' 和 stacked=True 参数用于创建堆叠条形图。
  5. 标签轴:记住标记 x 和 y 轴。

示例代码:

import pandas as pd
import matplotlib.pyplot as plt

# Create DataFrame from CSV data
df = pd.read_csv('data.csv')

# Restructure data
df2 = df.groupby(['Site Name', 'Abuse/NFF'])['Site Name'].count().unstack('Abuse/NFF').fillna(0)

# Create bar chart
df2[['abuse', 'nff']].plot(kind='bar', stacked=True)
plt.xlabel('Site Name')
plt.ylabel('Count')
plt.title('Stacked Bar Chart of Abuse and NFF')
plt.show()

以上是当数据分成多列时,如何用 Pandas 绘制堆积条形图?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn