Python pandas 按条件列表过滤数据框列

Question

过滤数据框中多列的最佳方法是什么？

例如，我从我的数据中得到了这个样本：

Index   Tag_Number
666052  A1-1100-XY-001
666382  B1-001-XX-X-XX
666385  **FROM** C1-0001-XXX-100
666620  D1-001-XX-X-HP some text

“Tag_Number”列包含标签，但我需要删除标签之前或之后的文本。 常见的分隔符是“空格”。 我的想法是将列分成多个并过滤以这些“A1-、B1-、C1-、D1-”中的任何一个开头的列中的每一列，即如果单元格不以条件开头，则为假，否则为真和应用到表，所以真值保持原样，但如果为假，我们得到空值。 最后，清理完标签后，将它们合并为一列。 我知道这可能很复杂，我非常愿意接受任何建议。

我已经尝试过的：

Splitted = df.Tag_Number.str.split(" ",expand=True)
Splitted.columns = Splitted.columns.astype(str)
Splitted = Splitted.rename(columns=lambda s: "Tag"+s)
col_names = list(Splitted.columns)
Splitted

我把这个 Tag_number 列分成 30 列，但现在我正在努力过滤掉每一列。 我创建了一个条件来过滤每一列：

asset = ('A1-','B1-','C1-','D1-')

但这并没有帮助，我只得到了最后一列的数组，而不是我猜的所有预期。

for col in col_names:
    Splitted_filter =  Splitted[col].str.startswith(asset, na = False)
Splitted_filter

有没有办法通过这个“资产”过滤器过滤每一列？

非常感谢

Answer 1

如果您想清除与资产前缀不匹配的文本，那么我认为这会起作用。

sample = pd.read_csv(StringIO("""Index,Tag_Number
666052,A1-1100-XY-001
666382,B1-001-XX-X-XX
666385,**FROM** C1-0001-XXX-100
666620,D1-001-XX-X-HP some text"""))

asset = ('A1-','B1-','C1-','D1-')
def asset_filter(tag_n):
    tags = tag_n.split() # common delimeter is "space"
    tags = [t for t in tags if len([a for a in asset if t.startswith(a)]) >= 1]
    return tags # can " ".join(tags) if str type is desired

sample['Filtered_Tag_Number'] = sample.Tag_Number.astype(str).apply(asset_filter)

看到可以定义自定义函数asset_filter并将其应用于您希望转换的列。

结果是这样的：

    Index   Tag_Number  Filtered_Tag_Number
0   666052  A1-1100-XY-001  [A1-1100-XY-001]
1   666382  B1-001-XX-X-XX  [B1-001-XX-X-XX]
2   666385  **FROM** C1-0001-XXX-100    [C1-0001-XXX-100]
3   666620  D1-001-XX-X-HP some text    [D1-001-XX-X-HP]