基于特定ID掩码拆分数据到两个CSV的Numpy技术问题求助

嘿，你在Geo Python课程的数据处理中卡在最后一步啦！我来帮你找出问题所在，然后给出修正方案～

首先先看你写的这段有问题的代码：

kmask = data[station==29980]
rmask = data[station==28450]
kumpula = kumpula[kmask]
rovaniemi = rovaniemi[rmask]

这里有几个小问题：

1. 掩码创建错误：station == 29980本身就是一个布尔类型的掩码数组，不需要用data[]去索引它，这样得到的不是掩码，而是直接提取了data里符合条件的行，和你要的逻辑不符。
2. 变量未定义就使用：kumpula和rovaniemi在赋值之前根本没定义，这肯定会报错。
3. 站点编号笔误：你之前说Rovaniemi对应station==29450，但代码里写的是28450，这会导致提取不到正确的数据哦。

正确的实现步骤（基于你前两步已完成的过滤） #

假设你前两步已经用tempmask过滤了station、date、temp_max、temp_min这几个数组，现在可以按下面的步骤来：

1. 创建正确的站点掩码
   直接基于过滤后的station数组生成布尔掩码：

   # 创建Kumpula站点的掩码（station编号29980）
   kmask = station == 29980
   # 创建Rovaniemi站点的掩码（station编号29450，修正笔误）
   rmask = station == 29450
   

2. 提取对应站点的所有数据列
   用掩码分别过滤每个变量，然后把它们组合成一个二维数组（方便导出CSV）：

   # 提取Kumpula的所有数据
   kumpula_data = np.column_stack([station[kmask], date[kmask], temp_max[kmask], temp_min[kmask]])
   # 提取Rovaniemi的所有数据
   rovaniemi_data = np.column_stack([station[rmask], date[rmask], temp_max[rmask], temp_min[rmask]])
   

3. 导出到CSV文件
   使用np.savetxt函数导出，记得设置表头、分隔符，以及保留合适的格式：

   # 导出Kumpula数据到CSV
   np.savetxt('kumpula_data.csv', kumpula_data, delimiter=',', header='station,date,temp_max,temp_min', fmt='%d,%d,%.1f,%.1f', comments='')
   # 导出Rovaniemi数据到CSV
   np.savetxt('rovaniemi_data.csv', rovaniemi_data, delimiter=',', header='station,date,temp_max,temp_min', fmt='%d,%d,%.1f,%.1f', comments='')
   

补充说明 #

• np.column_stack用来把多个一维数组按列组合成二维数组，这样导出的CSV结构更清晰。
• fmt参数用来指定每一列的格式：%d是整数（station和date），%.1f是保留一位小数的浮点数（温度值），你可以根据实际需求调整小数位数。
• comments=''是为了避免表头前面自动加上#号，让CSV的表头更规范。

这样修改后，你的代码就能正常运行，顺利导出两个站点的数据啦！

内容的提问来源于stack exchange，提问作者mkw