在Python中将CSV文件读取到多个NumPy数组

我想将包含各种股票价格的.csv文件导入到getData（）函数内部的Python脚本中，但我遇到了索引问题，无法看到如何解决问题。在Python中将CSV文件读取到多个NumPy数组

我是新来的CSV和NumPy的都这样我不确定是哪里的问题恰好，但是当我尝试运行此代码我收到以下内容：

文件“../StockPlot.py”，行20，在的getData 日期[I-1] =数据[0] IndexError：索引0超出范围为0轴与大小0

import numpy as np 
import matplotlib.pyplot as plt 
import csv 

def getData(): 
    date = np.array([]) 
    openPrice = np.array([]) 
    closePrice = np.array([]) 
    volume = np.array([]) 

    i = 1 
    with open('aapl.csv', 'rb') as f: 
     reader = csv.reader(open('aapl.csv')) 
     data_as_list = list(reader) 
     items = len(data_as_list) 

     while i < items: 
      data = data_as_list[i] 
      date[i-1] = data[0] 
      openPrice[i-1] = data[1] 
      closePrice[i-1] = data[4] 
      volume[i-1] = data[5] 
      i += 1 

    return date, openPrice, closePrice, volume 

getData()

我试图读取的文件AAPL.csv具有线形式：

Date, Open, High, Low, Close, Volume

26-Jul-17,153.35,153.93,153.06,153.46,15415545

25-Jul-17,151.80,153.84,151.80,152.74,18853932

24-Jul-17,150.58,152.44,149.90,152.09,21493160

我将不胜感激解决这个问题的任何帮助，似乎data_as_list是每行的列表的列表，并在玩打印功能后，似乎是打印数据[0]等循环，但不会让我的值分配给我创造

来源

2017-07-27 Euclidean

阵列IMO它更方便使用熊猫为：

import pandas as pd 

fn = r'/path/to/AAPL.csv'  
df = pd.read_csv(fn, skipinitialspace=True, parse_dates=['Date'])

结果：

In [83]: df 
Out[83]: 
     Date Open High  Low Close Volume 
0 2017-07-26 153.35 153.93 153.06 153.46 15415545 
1 2017-07-25 151.80 153.84 151.80 152.74 18853932 
2 2017-07-24 150.58 152.44 149.90 152.09 21493160

As numpy 2D array：

In [84]: df.values 
Out[84]: 
array([[Timestamp('2017-07-26 00:00:00'), 153.35, 153.93, 153.06, 153.46, 15415545], 
     [Timestamp('2017-07-25 00:00:00'), 151.8, 153.84, 151.8, 152.74, 18853932], 
     [Timestamp('2017-07-24 00:00:00'), 150.58, 152.44, 149.9, 152.09, 21493160]], dtype=object)

来源

2017-07-27 17:45:20 MaxU

在Python中将CSV文件读取到多个NumPy数组

回答

相关问题