用代码在原有csv中加入列风速数据原有的csv数据中是风速的垂直和水平数据

假设原有的CSV文件名为"original.csv"，其中包含垂直和水平风速数据的列分别为"VerticalSpeed"和"HorizontalSpeed"，需要添加的新列为"WindSpeed"，可以使用以下Python代码实现：

import csv
import math

# 打开原始CSV文件
with open('original.csv', 'r') as original_file:
    reader = csv.DictReader(original_file)
    # 获取原始数据的列名
    fieldnames = reader.fieldnames
    # 添加新的列名
    fieldnames.append('WindSpeed')
    # 打开新的CSV文件
    with open('new.csv', 'w', newline='') as new_file:
        writer = csv.DictWriter(new_file, fieldnames=fieldnames)
        writer.writeheader()
        # 逐行读取原始数据，计算风速并写入新的CSV文件
        for row in reader:
            v_speed = float(row['VerticalSpeed'])
            h_speed = float(row['HorizontalSpeed'])
            wind_speed = math.sqrt(v_speed ** 2 + h_speed ** 2)
            row['WindSpeed'] = wind_speed
            writer.writerow(row)

该代码会读取原始CSV文件，计算每行的风速并将结果写入新的CSV文件中。新的CSV文件中包含原始数据的所有列以及新的"WindSpeed"列。每行的"WindSpeed"值是通过计算垂直和水平风速数据得到的风速值。