Python pandas，多行绘图选项

Question

我想从熊猫数据框中绘制多条线，并为每条线设置不同的选项。我想做类似的事情

testdataframe=pd.DataFrame(np.arange(12).reshape(4,3))
testdataframe.plot(style=['s-','o-','^-'],color=['b','r','y'],linewidth=[2,1,1])

这将引发一些错误消息：

• 线宽不能用列表调用

• 在样式中，当在列表中定义颜色时，我不能使用’s’和’o’或任何其他字母符号

另外还有一些对我来说似乎很奇怪的东西

• 当我在上面的代码中添加另一个plot命令时，testdataframe[0].plot()它将在同一图中绘制该行，如果我添加命令testdataframe[[0,1]].plot()，它将创建一个新图

• 如果我愿意的话testdataframe[0].plot(style=['s-','o-','^-'],color=['b','r','y'])，可以使用样式列表，但不能使用颜色列表

希望有人可以帮助，谢谢。

Answer 1

你好亲密！

您可以在样式列表中指定颜色：

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

testdataframe = pd.DataFrame(np.arange(12).reshape(4,3), columns=['A', 'B', 'C'])
styles = ['bs-','ro-','y^-']
linewidths = [2, 1, 4]
fig, ax = plt.subplots()
for col, style, lw in zip(testdataframe.columns, styles, linewidths):
    testdataframe[col].plot(style=style, lw=lw, ax=ax)

还要注意，该plot方法可以使用一个matplotlib.axes对象，因此您可以进行多次html" target="_blank">调用（如果需要）：

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

testdataframe1 = pd.DataFrame(np.arange(12).reshape(4,3), columns=['A', 'B', 'C'])
testdataframe2 = pd.DataFrame(np.random.normal(size=(4,3)), columns=['D', 'E', 'F'])
styles1 = ['bs-','ro-','y^-']
styles2 = ['rs-','go-','b^-']
fig, ax = plt.subplots()
testdataframe1.plot(style=styles1, ax=ax)
testdataframe2.plot(style=styles2, ax=ax)

在这种情况下，这不是很实际，但是以后可能会派上用场。