Matplotlib绘制股票K线图添加文本时报错的解决求助

一、TypeError: only size-1 arrays can be converted to Python scalars 错误修复 #

这个错误百分百是你在plt.text()里传了数组/Series类型参数——plt.text要求x、y坐标和文本内容必须是单个标量值，不能直接把整列数据塞进去。

举个典型错误写法：

# 假设date、close是pandas Series
plt.text(date, close, str(close))  # 直接传Series，触发类型错误

正确做法是遍历每个元素，逐个传入标量：

# 先把日期转成Matplotlib能识别的数值（如果x轴是日期）
df['date_num'] = mdates.date2num(df['date'])

# 循环遍历每一组坐标和文本
for x, price in zip(df['date_num'], df['close']):
    plt.text(x, price + 0.2, f'{price:.2f}', fontsize=7, ha='center')

二、the truth value of a series is ambiguous 错误预防 #

这个错误出现在直接用Series做条件判断的场景（比如if df['close'] > df['open']:），因为布尔Series不能直接作为if的判断条件。

K线涨跌颜色判断应该用向量化操作，避免循环和错误判断：

import numpy as np
# 一行代码搞定涨跌颜色赋值
df['color'] = np.where(df['close'] > df['open'], 'red', 'green')

三、图片保存报错的常见修复 #

保存图片报错大概率是这几个原因：

• 路径问题：用绝对路径，避免相对路径混乱，同时确保目录存在

  plt.savefig('D:/trading/stock_kline.png')
  

• 画布清空问题：plt.show()会清空画布，必须在plt.show()之前调用plt.savefig()
• 内容截断：保存前加plt.tight_layout()，避免图表元素被截断
• 权限问题：别往系统保护目录（比如C盘根目录）存，选有写入权限的文件夹

四、可运行的简化版K线图代码 #

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

# 读取数据（假设你的CSV包含date、open、close、high、low列）
df = pd.read_csv('stock_data.csv')
df['date'] = pd.to_datetime(df['date'])
df['date_num'] = mdates.date2num(df['date'])

# 计算K线实体和影线参数
df['color'] = np.where(df['close'] > df['open'], '#ff4500', '#32cd32')
df['body_top'] = np.maximum(df['open'], df['close'])
df['body_height'] = abs(df['close'] - df['open'])

# 绘图
fig, ax = plt.subplots(figsize=(12, 6))
# 画K线实体
ax.bar(df['date_num'], df['body_height'], bottom=df['body_top']-df['body_height'], color=df['color'], width=0.6)
# 画上下影线
ax.vlines(df['date_num'], df['low'], df['high'], color=df['color'], linewidth=1)

# 设置日期格式
ax.xaxis.set_major_formatter(mdates.DateFormatter('%Y-%m-%d'))
plt.xticks(rotation=45)

# 循环添加收盘价文本
for x, close in zip(df['date_num'], df['close']):
    ax.text(x, close + 0.3, f'{close:.2f}', fontsize=7, ha='center')

# 保存并展示
plt.tight_layout()
plt.savefig('stock_kline.png')
plt.show()

内容的提问来源于stack exchange，提问作者Amg91