在 python 中读取 csv 文件时附加到混合的二维数组

Question

我有一个逗号分隔的 csv 文件“Brent 3.csv”： 行如下所示：

2014.03.12 23:59:59,2014.03.20 23:59:59,BRENTSPOT,Brent 
1,1.29,1.6,0.8568833439015613,91.09,3.5,2.053,-0.035\n

2014.04.01 23:59:59,2014.04.02 23:59:59,BRENTSPOT,Brent 
1,1.39,1.4,0.8568833439015613,89.59999999999999,3.5,2.053,-0.036\n

现在我想做的是创建一个包含字符串和浮点值的二维混合列表。

import os

def create_list(stratname,directory):
    os.chdir(directory)
    temp = []

    for file in glob.glob("*.csv"):
        if stratname in file:
            TDFile=open(file,"r")

            for i,line in enumerate(TDFile):
                s = line.split(',')
                for x in s:   
                    try:
                        temp.append(float(x))
                    except ValueError:
                        temp.append(x)
    return temp


brent3 = create_list("Brent 3",strategydir)

print(brent3)

我知道我应该使用

temp.append([float(x)])

但这只会创建一个看起来像这样的列表：

['2014.03.12 23:59:59', '2014.03.20 23:59:59', 'BRENTSPOT', 'Brent 1', [1.29], [1.6], [0.8568833439015613], [91.09], [3.5], [2.053], [-0.035], '2014.04.01 23:59:59',    ...   ]

应该是：

[['2014.03.12 23:59:59', '2014.03.20 23:59:59', 'BRENTSPOT', 'Brent 1', 1.29, 1.6, 0.8568833439015613, 91.09, 3.5, 2.053, -0.035], ['2014.04.01 23:59:59', .....]]

我似乎无法找到答案如何以 [[first line],[second line]] 格式获取 csv 文件的每一行。

任何帮助表示赞赏:)

Answer 1

def create_list(stratname,directory):
    os.chdir(directory)
    temp = []
    fin = []

    for file in glob.glob("*.csv"): # search file
        if stratname in file:
            TDFile=open(file,"r")

            for line in TDFile: # go throug every line in file
                temp = []
                s = line.split(',')
                for x in s:
                    try:
                        temp.append(round(float(x),3))
                    except ValueError:
                        temp.append(x)
                fin.append(temp)
    return fin

这解决了问题。我添加了另一个数组 fin = []，我将临时值附加到每一行并再次清空 temp = [] 中的值以用于文件中的下一行。

Answer 2

为什么不使用pandas：

import io
import pandas as pd

csv = """2014.03.12 23:59:59,2014.03.20 23:59:59,BRENTSPOT,Brent 1,1.29,1.6,0.8568833439015613,91.09,3.5,2.053,-0.035\n
2014.04.01 23:59:59,2014.04.02 23:59:59,BRENTSPOT,Brent 1,1.39,1.4,0.8568833439015613,89.59999999999999,3.5,2.053,-0.036\n"""

print(pd.read_csv(io.StringIO(csv), header=None).values.tolist())

产生

[['2014.03.12 23:59:59', '2014.03.20 23:59:59', 'BRENTSPOT', 'Brent 1', 1.29, 1.6, 0.8568833439015613, 91.09, 3.5, 2.053, -0.035],
 ['2014.04.01 23:59:59', '2014.04.02 23:59:59', 'BRENTSPOT', 'Brent 1', 1.39, 1.4, 0.8568833439015613, 89.59999999999998, 3.5, 2.053, -0.036000000000000004]]

如果您想使用您的文件 Brent 3.csv，请将对 io.StringIO 的调用替换为文件的路径。