Perl实现按气象参数拆分数据并写入对应文件的技术咨询

需求说明 #

我有一组包含地点名称和4个气象参数（24小时降雨量H1dR、5日总降雨量H5dR、最低气温Tmin、最高气温Tmax）对应风险等级（1-4）的数据。单日所有参数均为等级1（无风险）时无需处理，否则要处理1到4个参数。我已经能判断是否无需处理，也能通过文件头生成包含需处理气象元素的@RiskArray数组。现在需要实现拆分每行数据，为每个气象元素单独生成文件，文件中只保留地点和对应风险值两列。现有一段Perl代码，但不知道while循环里该怎么写，相关代码和输出样例如下：

原代码片段 #

#!/usr/bin/perl -w

use strict;
use warnings;

my @RiskArray=("H1dR", "H5dR", "Tmin", "Tmax");
my $arraynum=scalar @RiskArray;

my $hdr=<DATA>;
for my $i (0..$#RiskArray){
 while (my $line = <DATA>){

 }
}

__DATA__
Risk    H1dR  H5dR  Tmin  Tmax
Maputsoe, 2, 1, 3,  2
Butha-Buthe (Butha-Buthe District), 4, 3, 2, 1
Mohale's Hoek, 2, 2, 3, 4
Qacha's Nek, 2, 3, 1,  2

预期输出示例 #

H1dRrv.txt内容

Maputsoe, 2
Butha-Buthe (Butha-Buthe District), 4
Mohale's Hoek, 2
Qacha's Nek, 2

H5dRrv.txt内容

Maputsoe, 1
Butha-Buthe (Butha-Buthe District), 3
Mohale's Hoek, 2
Qacha's Nek, 3

解决方案 #

原代码的问题在于外层for循环和内层while循环的嵌套会导致第一次循环就读完所有数据行，后续循环无法获取数据。以下是修正后的完整代码：

#!/usr/bin/perl -w
use strict;
use warnings;

my @RiskArray = ("H1dR", "H5dR", "Tmin", "Tmax");
# 存储每个参数对应的文件句柄
my %file_handles;

# 为每个参数创建输出文件并打开句柄
foreach my $param (@RiskArray) {
    open my $fh, '>', "${param}rv.txt" or die "无法创建文件 ${param}rv.txt: $!";
    $file_handles{$param} = $fh;
}

# 跳过表头行
my $hdr = <DATA>;

# 逐行处理数据
while (my $line = <DATA>) {
    chomp $line;
    # 拆分地点和风险值：兼容地点含特殊字符的情况
    if ($line =~ /^(.*?),\s*(\d+),\s*(\d+),\s*(\d+),\s*(\d+)$/) {
        my $location = $1;
        my @risks = ($2, $3, $4, $5);
        
        # 将对应参数的风险值写入文件
        for my $i (0..$#RiskArray) {
            my $fh = $file_handles{$RiskArray[$i]};
            print $fh "$location, $risks[$i]\n";
        }
    }
}

# 关闭所有文件句柄
foreach my $fh (values %file_handles) {
    close $fh;
}

__DATA__
Risk    H1dR  H5dR  Tmin  Tmax
Maputsoe, 2, 1, 3,  2
Butha-Buthe (Butha-Buthe District), 4, 3, 2, 1
Mohale's Hoek, 2, 2, 3, 4
Qacha's Nek, 2, 3, 1,  2

关键逻辑说明 #

1. 预先创建文件句柄：先为每个气象参数生成对应的输出文件（如H1dRrv.txt），并将文件句柄存入哈希表，避免重复打开/关闭文件。
2. 数据拆分：使用正则表达式^(.*?),\s*(\d+),\s*(\d+),\s*(\d+),\s*(\d+)$拆分每行数据，确保地点部分（可能包含特殊字符）被完整提取，最后四个数值为对应参数的风险等级。
3. 逐行写入：遍历每行数据，将每个参数的风险值写入对应文件，确保数据一一对应。
4. 关闭文件句柄：处理完所有数据后统一关闭文件句柄，保证数据完整写入磁盘。

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