如何拆分蜡烛图对象列表以调用matplotlib的candlestick2_ohlc绘制OHLC图

嗨，我来帮你搞定这个问题！要把你的Candle对象列表转换成candlestick2_ohlc需要的四个数组（opens、highs、lows、closes），其实用Python的基础语法就能轻松实现，要是需要更高效的数值操作，也可以结合numpy来处理。下面给你详细的步骤和完整示例：

方法一：用列表推导式生成普通Python列表 #

这是最直观的方式，直接遍历你的Candle对象列表，逐个提取对应属性即可：

# 假设你的Candle对象列表名为candles_list
opens = [candle.open for candle in candles_list]
highs = [candle.high for candle in candles_list]
lows = [candle.low for candle in candles_list]
closes = [candle.close for candle in candles_list]

这种方式代码简洁，适合数据量不大的场景，完全能满足绘图需求。

方法二：转换为numpy数组（推荐用于后续数值计算） #

如果之后还要对数据做统计、分析等操作，转成numpy数组会更高效：

import numpy as np

# 先把所有OHLC数据整合为二维数组，再拆分
ohlc_data = np.array([[c.open, c.high, c.low, c.close] for c in candles_list])
opens = ohlc_data[:, 0]
highs = ohlc_data[:, 1]
lows = ohlc_data[:, 2]
closes = ohlc_data[:, 3]

完整绘图示例代码 #

下面是包含Candle类定义、测试数据、拆分逻辑和绘图的完整代码：

import matplotlib.pyplot as plt
from matplotlib.finance import candlestick2_ohlc
import numpy as np

# 你的Candle类定义
class Candle:
    def __init__(self, open, high, low, close, volume, timestamp):
        self.open = open
        self.high = high
        self.low = low
        self.close = close
        self.volume = volume
        self.timestamp = timestamp

# 创建测试用的Candle对象列表
candles_list = [
    Candle(100, 105, 98, 103, 1000, 1620000000),
    Candle(103, 108, 101, 106, 1200, 1620086400),
    Candle(106, 107, 102, 104, 900, 1620172800),
    Candle(104, 109, 103, 108, 1500, 1620259200),
    Candle(108, 112, 107, 110, 1800, 1620345600)
]

# 拆分OHLC数据（这里用列表推导式示例）
opens = [c.open for c in candles_list]
highs = [c.high for c in candles_list]
lows = [c.low for c in candles_list]
closes = [c.close for c in candles_list]

# 绘制蜡烛图
fig, ax = plt.subplots()
candlestick2_ohlc(ax, opens, highs, lows, closes, width=4, colorup='k', colordown='r', alpha=0.75)

# 添加基础样式
ax.set_title('OHLC Candlestick Chart')
ax.set_xlabel('Candle Index')
ax.set_ylabel('Price')
plt.show()

小提示：如果你用的是新版本matplotlib，matplotlib.finance模块已经被迁移到第三方库mplfinance，这个库的API更简洁且维护更活跃，你可以考虑切换过去，但如果坚持使用原函数，上面的代码依然有效。

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