ax.set_title(df['name'].iloc[0])
时间: 2023-10-07 09:03:58 浏览: 39
是的,这个代码可以将图表的标题设置为该数据框中第一行的股票名称。如果你的数据框包含多个股票,则需要分别为每个数据框设置标题。你可以使用一个循环来绘制多个图表,并为每个图表设置相应的标题。以下是一个示例代码,可以绘制多个股票的K线图,并将它们的名称作为标题:
```
import matplotlib.pyplot as plt
import matplotlib.dates as mdates
import pandas as pd
def plot_k_line(df):
df['date'] = pd.to_datetime(df['date'])
df['date'] = df['date'].apply(mdates.date2num)
fig, ax = plt.subplots()
ax.plot(df['date'], df['close'], label='Close')
ax.plot(df['date'], df['open'], label='Open')
ax.plot(df['date'], df['high'], label='High')
ax.plot(df['date'], df['low'], label='Low')
ax.legend()
ax.set_xlabel('Date')
ax.set_ylabel('Price')
ax.set_title(df['name'].iloc[0])
# 创建一个包含多个数据框的列表
stock_data = [df1, df2, df3, ...]
# 循环绘制每个股票的K线图,并设置相应的标题
for df in stock_data:
plot_k_line(df)
plt.show()
```
其中,`stock_data`是一个包含多个数据框的列表,每个数据框包含一个股票的K线数据。在循环中,我们依次使用每个数据框调用`plot_k_line`函数,并使用该数据框的第一行的名称设置图表的标题。注意,我们在每次循环结束后使用`plt.show()`显示图表,以便在绘制完一个股票后,再绘制下一个股票。