3

我有一个包含 38 列和 1500 多行的 csv，其中包含浮点数和字符串。我想要这个集合中的 3 列（x，y，z）浮点数据来找到f=(x+y)/z. 经过研究，我成功地将这些列隔离为 numpy 数组并执行f=(x+y)/z. 现在，当我尝试对 f 求和时，数组没有加起来。我打印 f 我看到 1500 项正确值，但不是这些值的总和。

  reader=csv.reader(open('myfile.csv' ,"rb"),delimiter=',')
  reader.next()
  reader.next()
  x=list(reader)
  data=numpy.array(x)
  rows=data.shape[0]
  for i in range (0,rows):
      x=numpy.array(data[i,18]).astype('float')
      y=numpy.array(data[i,19]).astype('float')
      z=numpy.array(data[i,6]).astype('float')
      f=numpy.array((x+y)/z)
      average=numpy.sum(f)/rows
      print average
4

3 回答 3

5

如果data已经是一个数组，则不需要for循环：

x = data[:, 18].astype(float)
y = data[:, 19].astype(float)
z = data[:, 6].astype(float)
f = (x+y) / z
average = np.average(f)

阅读您的文件可能会更好np.loadtxt：

data = np.loadtxt('myfile.csv', dtype=float, delimiter=',' skiprows=2,
                  usecols=(6, 18, 19))

或得到x,y并z直接：

x, y, z = np.loadtxt('myfile.csv', dtype=float, delimiter=',' skiprows=2,
                     usecols=(6, 18, 19), unpack=True)
于 2013-03-16T15:09:07.750 回答
2

Numpy 允许您将数组作为一个整体进行操作，您不需要遍历它们。

reader=csv.reader(open('myfile.csv' ,"rb"),delimiter=',')
reader.next()
reader.next()
x=list(reader)
data=numpy.array(x)
rows=data.shape[0]

x=data[:,18].astype('float')
y=data[:,19].astype('float')
z=data[:,6].astype('float')

f = (x + y) / z
average = f.mean()

print average
于 2013-03-16T15:13:11.740 回答
0

如果您没有锁定 numpy，这里是一个纯 python 解决方案，

import csv

def f(x, y, z):
    x = float(x)
    y = float(y)
    z = float(z)
    return (x+y)/z

reader = csv.reader(open("derp.csv", 'r'))
rows = list(reader)
len_of_rows = len(rows)

f_values = []

for row in rows:
    x = row[0]
    y = row[1]
    z = row[2]
    f_values.append(f(x, y, z))

average = sum(f_values)/len_of_rows
print average

这是我的 derp.csv 的样子

1,2,3
4,5,6
7,8,9
于 2013-03-16T15:05:57.660 回答