在字符串列中删除单词并替换字符

Question

我需要更改已导入的数据帧的DSFS列中的值。你知道吗

MemberID,Year,DSFS,DrugCount
48925661,Y2,9-10 months,7+
90764620,Y3,8- 9 months,3
61221204,Y1,2- 3 months,1

例如，“9-10个月”需要改为9-10。你知道吗

我该怎么做？你知道吗

Answer 1

我没有安装pandas，但是解决方案应该适用于df对象。你知道吗

string="48925661,Y2,9-10 months,7+"
"_".join(re.findall(r'\b\d+\b', string.split(",")[2]))

试验结果：

>>> "_".join(re.findall(r'\b\d+\b', string.split(",")[2]))
'9_10'

python脚本：

$ cat test.py
with open("sample.csv") as inputs:
    next(inputs)  # skip the first line
    for line in inputs:
        parts = line.strip().split(",")
        parts[2] = "_".join(re.findall(r'\b\d+\b', parts[2]))
        print(",".join(parts))

结果：

$python test.py                                
48925661,Y2,9_10,7+
90764620,Y3,8_9,3
61221204,Y1,2_3,1