Python - .split(), - 两个参数

Question

我需要对 python 代码进行修改。

此代码从 .csv 文件中抓取信息，最终以不同的结构将其集成到新的 .csv 文件中。

在源文件的一列中，我有一个值（字符串），99% 的时间都是这样形成的：'block1 block2 block3'。

Block2 在 99% 的情况下总是以值 'm' 结束。

示例：“R2 180m RFT”。

通过浏览源数据集，我意识到在 1% 的情况下，block2 可以以“M”结尾。

因为我需要 'm' 或 'M' 值之后的所有值，所以我有点卡住了。

我使用了 .split() 函数，就像在我的 :

'Newcolumn': getattr(row_unique_ids, 'COLUMNINTHEDATASET').split ('m') [1],

这样做，我的脚本出错了，因为它落在了这种风格的值上： R2 180M AST'。

所以我想知道如何集成一个额外的参数，如果脚本落在“m”或“M”上，这将使我能够很好地进行拆分。

感谢您的帮助。

Answer 1

一个解决方案是

s = getattr(row_unique_ids, 'COLUMNINTHEDATASET')
s = s.lower()
s.split('m')[1]

但这会弄乱你的外壳。如果要保存外壳， 另一种解决方案是：

x = ''
s = getattr(row_unique_ids, 'COLUMNINTHEDATASET')
for c in s:
  if c == 'M'
    x += 'm'
  x += c
x.split('m')[1]

Answer 2

进行多参数拆分的一种方法通常是：

import re

string = "this is 3an infamous String4that I need to s?plit in an infamou.s way"

#Preserve the original char
print (re.sub(r"([0-9]|[?.]|[A-Z])",r'\1'+"DELIMITER",string).split('DELIMITER'))

#Discard the original char
print (re.sub(r"([0-9]|[?.]|[A-Z])","DELIMITER",string).split('DELIMITER'))

输出：

['this is 3', 'an infamous S', 'tring4', 'that I', ' need to s?', 'plit in an infamou.', 's way']
['this is ', 'an infamous ', 'tring', 'that ', ' need to s', 'plit in an infamou', 's way']

在您的上下文中：

import re

string = "R2 180m RFT R2 180M RFT"

print (re.sub(r"\b([0-9]+)[mM]\b",r'\1'+"M",string).split('M'))
#print (re.sub(r"\b([0-9]+)[mM]\b",r'\1'+"M",getattr(row_unique_ids, 'COLUMNINTHEDATASET')).split('M'))

输出：

['R2 180', ' RFT R2 180', ' RFT']

如果m 和M 前面有数字，它将分开。