pandas作各种图的完整代码

这里提供pandas绘制常见图形的完整代码示例。

折线图

import pandas as pd
import matplotlib.pyplot as plt

# 创建数据
data = {
    'year': [2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020],
    'sales': [100, 200, 500, 800, 900, 1000, 1200, 1500, 1800, 2000, 2200]
}
df = pd.DataFrame(data)

# 绘制折线图
df.plot(x='year', y='sales', kind='line', title='Sales Trend')

# 显示图形
plt.show()

柱状图

import pandas as pd
import matplotlib.pyplot as plt

# 创建数据
data = {
    'name': ['Tom', 'Jerry', 'Mike', 'Lucy', 'Bob'],
    'score': [80, 90, 70, 85, 92]
}
df = pd.DataFrame(data)

# 绘制柱状图
df.plot(x='name', y='score', kind='bar', title='Scores')

# 显示图形
plt.show()

散点图

import pandas as pd
import matplotlib.pyplot as plt

# 创建数据
data = {
    'height': [170, 175, 180, 185, 190],
    'weight': [65, 70, 75, 80, 85]
}
df = pd.DataFrame(data)

# 绘制散点图
df.plot(x='height', y='weight', kind='scatter', title='Height vs Weight')

# 显示图形
plt.show()

饼图

import pandas as pd
import matplotlib.pyplot as plt

# 创建数据
data = {
    'category': ['A', 'B', 'C', 'D'],
    'value': [20, 30, 25, 15]
}
df = pd.DataFrame(data)

# 绘制饼图
df.plot(kind='pie', y='value', labels=df['category'], autopct='%1.1f%%', title='Category Distribution')

# 显示图形
plt.show()

箱线图

import pandas as pd
import matplotlib.pyplot as plt

# 创建数据
data = {
    'group': ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
    'value': [10, 20, 30, 15, 25, 35, 20, 30, 40]
}
df = pd.DataFrame(data)

# 绘制箱线图
df.boxplot(column='value', by='group')

# 显示图形
plt.show()

区域图

import pandas as pd
import matplotlib.pyplot as plt

# 创建数据
data = {
    'year': [2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020],
    'sales': [100, 200, 500, 800, 900, 1000, 1200, 1500, 1800, 2000, 2200]
}
df = pd.DataFrame(data)

# 绘制区域图
df.plot(x='year', y='sales', kind='area', title='Sales Trend')

# 显示图形
plt.show()

热力图

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

# 创建数据
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [6, 7, 8, 9, 10],
    'C': [11, 12, 13, 14, 15],
    'D': [16, 17, 18, 19, 20],
    'E': [21, 22, 23, 24, 25]
}
df = pd.DataFrame(data)

# 计算相关系数矩阵
corr = df.corr()

# 绘制热力图
plt.imshow(corr, cmap='hot', interpolation='none')
plt.colorbar()
plt.xticks(range(len(corr)), corr.columns)
plt.yticks(range(len(corr)), corr.columns)
plt.title('Correlation Matrix')

# 显示图形
plt.show()

以上是pandas绘制常见图形的完整代码示例，可以根据需要进行修改和扩展