使用pandas和matplotlib创建具有不同颜色的单个条形图

Question

我有两个dfs，我想创建一个单独的条形图， 每个酒吧都需要自己的颜色，具体取决于它来自哪个。

# Ages < 20
df1.tags = ['locari', 'ママコーデ', 'ponte_fashion', 'kurashiru', 'fashion']
df1.tag_count = [2162, 1647, 1443, 1173, 1032]

# Ages 20 - 24
df2.tags= ['instagood', 'ootd', 'fashion', 'followme', 'love']
df2.tag_count = [6523, 4576, 3986, 3847, 3599]

如何创建这样的情节？

P.S。原来的df更大了。有些单词可能重叠，但我希望它们也有不同的颜色

Answer 1

您的数据框tag_counts只是简单的列表，因此您可以使用标准的mpl条形图在同一轴上绘制它们。这个答案假设两个数据帧具有相同的长度。

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np

# Create dataframes
df1=pd.DataFrame()
df2=pd.DataFrame()

# Ages < 20
df1.tags = ['locari', 'blub', 'ponte_fashion', 'kurashiru', 'fashion']
df1.tag_count = [2162, 1647, 1443, 1173, 1032]

# Ages 20 - 24
df2.tags= ['instagood', 'ootd', 'fashion', 'followme', 'love']
df2.tag_count = [6523, 4576, 3986, 3847, 3599]

# Create figure
fig=plt.figure()
ax=fig.add_subplot(111)

# x-coordinates
ind1 = np.arange(len(df1.tag_count))
ind2 = np.arange(len(df2.tag_count))
width = 0.35

# Bar plot for df1
ax.bar(ind1,df1.tag_count,width,color='r')

# Bar plot for df1
ax.bar(ind2+width,df2.tag_count,width,color='b')

# Create new xticks
ticks=list(ind1+0.5*width)+list(ind2+1.5*width)
ticks.sort()
ax.set_xticks(ticks)

# Sort labels in an alternating way
labels = [None]*(len(df1.tags)+len(df2.tags))
labels[::2] = df1.tags
labels[1::2] = df2.tags
ax.set_xticklabels(labels)

plt.show()

这将返回这样的情节

请注意，要将tags合并到一个列表中，我假设两个列表具有相同的长度。