专栏名称: 人工智能头条
专注人工智能技术前沿、实战技巧及大牛心得。
目录
相关文章推荐
黄建同学  ·  AI Agent ... ·  昨天  
人工智能产业链union  ·  【AI加油站】第十四部:《LLM ... ·  2 天前  
人工智能产业链union  ·  【AI加油站】第十四部:《LLM ... ·  2 天前  
51好读  ›  专栏  ›  人工智能头条

用Python绘制棒棒糖图表,真的好看!

人工智能头条  · 公众号  · AI  · 2021-05-25 18:00

正文

请到「今天看啥」查看全文



fig, ax = plt.subplots( 1 , figsize=( 12 , 8 ))

# 年份数
n = len(df)
# 颜色设置
colors = [
'black' ] + ((n -1 )*[ 'lightgrey' ])
plt.bar(df.Year, df.value, color=colors)
plt.scatter(df.Year, df.value, color=colors)
plt.show()

得到结果如下。

颜色已经修改成功,还需要调整一下条形图的宽度以及顶部圆圈的大小。

# width: 条形图宽度  s: 散点图圆圈大小
plt.bar(df.Year, df.value, color=colors, width=0.2)
plt.scatter(df.Year, df.value, color=colors, s=10)
plt.show()

结果如下。

比起先前的蓝色条形图图表,棒棒糖图表确实是好看了不少。

除了用条形图来绘制棒棒糖图表,还可以使用线条,这样整体的宽度会更加一致。

X将Year(年份)数据作为起点和终点,Y以-20和各年份数据作为起点和终点。

import pandas as pd
import matplotlib.pyplot as plt

# 读取数据
df = pd.read_csv('data.csv')
print(df)

# 新建画布
fig, ax = plt.subplots(1, figsize=(
128))

# 年份数
n = len(df)
# 颜色设置
colors = [
'black'] + ((n-1)*['lightgrey'])
# 使用线条
for idx, val in df.iterrows():
    plt.plot([val.Year, val.Year],
             [
-20, val.value],
             color=colors[idx])
plt.show()

得到结果如下。







请到「今天看啥」查看全文