python如何将CSV的数据读取为二维数组

Question

例如CSV里有三列数据分别为lon lat data 如何读取为以lon和lat为索引的data数组？

Answer 1

import csv

# 定义经度和纬度的范围以及每个经度、纬度的格点数
min_lon, max_lon = 100, 110
min_lat, max_lat = 20, 30
num_lon, num_lat = 11, 11

# 读取 CSV 数据
with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    # 跳过标题行
    next(reader)
    # 读取数据并存入二维数组
    data = [list(map(float, row)) for row in reader]

# 构造二维数组，并初始化为 0
data_array = [[0 for i in range(num_lat)] for j in range(num_lon)]
# 将数据填充到数组中
for row in data:
    lon_index = int(row[0] - min_lon)
    lat_index = int(row[1] - min_lat)
    data_array[lon_index][lat_index] = row[2]

print(data_array)