如何使用Python从.CSV文件中找到AWND的前10行，并将结果存储在新的.CSV文件中？

import csv

input_file = 'input.csv'
output_file = 'output.csv'

with open(input_file, 'r') as file:
    reader = csv.reader(file)
    data = list(reader)

header = data[0]
awnd_index = header.index('AWND')

sorted_data = sorted(data[1:], key=lambda x: float(x[awnd_index]))

top_10_rows = sorted_data[:10]

with open(output_file, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(header)
    writer.writerows(top_10_rows)

import csv

input_file = 'input.csv'
output_file = 'output.csv'

with open(input_file, 'r') as file:
    reader = csv.reader(file)
    data = list(reader)

header = data[0]
awnd_index = header.index('AWND')

sorted_data = sorted(data[1:], key=lambda x: float(x[awnd_index]))
top_10_rows = sorted_data[:10]

with open(output_file, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(header)
    writer.writerows(top_10_rows)