如何在一个DataFrame中绘制多个数据列
Python有很多有用的软件包,如pandas、matplotlib、numpy等。要使用数据框架,我们需要pandas库,要绘制数据框架的列,我们需要matplotlib。Pandas与Matplotlib紧密结合。你可以使用plot()方法直接从你的DataFrame中绘制数据。
要在单个框架中绘制多个数据列,我们只需将列的列表传递给绘图函数的y参数。下面是一个正确的方法,并附有实施示例。
步骤:
- Import module
- 创建或加载数据
- 转换为数据框架
- 使用plot()方法,沿X轴指定一列,沿Y轴指定多列作为数组。
- 显示图。
下面是几个例子,说明了上述在数据框架中绘制多列数据的方法。
示例 1:
import pandas as pd
import matplotlib.pyplot as mp
# take data
data = pd.read_csv("Bestsellers.csv")
# form dataframe
data = data.head()
df = pd.DataFrame(data, columns=["Name", "Price", "User Rating"])
# plot the dataframe
df.plot(x="Name", y=["Price", "User Rating"], kind="bar", figsize=(9, 8))
# print bar graph
mp.show()
输出:
示例 2:
import pandas as pd
import matplotlib.pyplot as mp
# data to be plotted
data = [["New York", 8.6, 20],
["Chicago", 2.7, 20],
["Los Angeles", 3.9, 20],
["Philadelphia", 1.5, 20],
["Houston", 2.1, 20]]
# form dataframe from data
df = pd.DataFrame(data, columns=["City", "Population(million)", "Year(2020)"])
# plot multiple columns such as population and year from dataframe
df.plot(x="City", y=["Population(million)", "Year(2020)"],
kind="line", figsize=(10, 10))
# display plot
mp.show()
输出: