首页 > 解决方案 > 如何从numpy数组中提取列

问题描述

我正在尝试制作一个机器人,它可以为我提供所需股票的图表。我yfinance为此使用 API,但问题是当我说它APPL.history为我提供以下 APPL 股票值时:“开盘价、低点、高点、收盘价、成交量、股息”,而我只对开盘价感兴趣。

这是我的代码:

import matplotlib
import numpy as np
import yfinance as yf


def stock_data(ticker='', period='', interval=''):
   ticker = yf.Ticker(ticker)
   ticker_history = ticker.history(period, interval)
   print(ticker_history)

def open_price(ticker, period='', interval=''):
    price = np.extract(open)
    print(price)

open_price('APPL','3mo','1wk')

标签: pythonnumpy

解决方案


假设你有一个ticker = yf.Ticker('AAPL')history_as_df = ticker.history()将返回一个 python 数据框的代码。

请查看10 分钟给 pandas以熟悉 pandas 数据框。

只选择Open列做

history_as_df[['Open']]

要将这些值提取为numpy数组,请执行

history_as_df[['Open']].values

推荐阅读