使用 Python 中的模块运行 CSV 的每一行

Question

在 Python 3.8.5 中工作

问题：CSV 文件包含 100 行，两列包含模块执行计算所需的数据。我想用两个数据点运行每一行，获取输出并将其插入第三列。

到目前为止的操作：我找到了 CSV 模块，可以使用 CSV.reader 读取每一行。我可以看到如何获取数据点的输出，但看不到如何获取它们并在我需要运行以处理数据的模块中使用它们。我还发现了子进程，我相信它是可以让我处理每一行的模块。我只是发现很难将这两者联系起来。

示例数据：

DateTime,Date,Time,Wind_Direction,Effective_Fetch,Wind_Speed
01/10/2012 00:00,01/10/2012,00:00:00,228,510,1.976
01/10/2012 00:10,01/10/2012,00:10:00,231,516,1.389
01/10/2012 00:20,01/10/2012,00:20:00,239,532,1.759

我要处理的两列是Effective_Fetch 和Wind_Speed。

模块如下：

def Hs(w, Lf):
    gravity=9.81 #ms^-2
    slope=0.0026
    x = (slope)*(gravity**(-0.53))*(w**(1.06)*(Lf**(0.47)))
    return x

w 是 Wind_Speed，Lf 是 Effective_Fetch 和 x 是我想插入到 Wind_Speed 后面的列中的值，列标题为“Wave_Height” - 我已经阅读在 Pandas 中也应该能够做到这一点的其他模块。

Answer 1

你可能想要这样的东西

output_rows = []
with open('mycsv.csv', newline='') as f:
    reader = csv.reader(f)
    # Skip the first (header) row
    headers = next(reader)
    for row in reader:
        # Unpack the two values we are interested in, ignore the others
        *_, effective_fetch, wind_speed = row
        # Values read from CSVs are strings, so cast them to numeric types
        result = Hs(float(wind_speed), int(effective_fetch))
        # Make a new row of the original row and the result of calling Hs
        output_rows.append(row + [result])

# Write out a new csv (if required)
with open('mynewcsv.csv', 'w', newline='') as new_f:
    writer = csv.writer(new_f)
    writer.writerow(headers + ['wave_height'])
    writer.writerows(output_rows)

csv.reader 对象是一个迭代器，因此使用next 函数将其向前推进一步。这比在主 for 循环中使用条件来检查我们是否正在处理第一行更尴尬。

Hs 函数需要两个输入，幸运的是它们是 csv 行中的最后两列。

*_, effective_fetch, wind_speed = row

告诉解释器将最后两列的值分配给effective_fetch和wind_speed，并将所有前面的列分配给一个名为_的变量，这是我们想要的变量的通用命名约定忽略（当然，你可以随意称呼它）。

你也可以通过行索引来做到这一点，特别是如果列的位置不太方便：

 effective_fetch, wind_speed = row[4], row[5]

或从末尾开始索引：

 effective_fetch, wind_speed = row[-2], row[-1]

或通过列表切片：

 effective_fetch, wind_speed = row[4:]
 effective_fetch, wind_speed = row[-2:]