时间戳

首页 > 解决方案 > 如果列名以数字开头，则删除 pandas 列

python - 如果列名以数字开头，则删除 pandas 列

问题描述

我有一个大约 200 列的 pandas DataFrame。粗略地说，我想这样做

for col in df.columns:
    if col begins with a number:
        df.drop(col)

我不确定处理 pandas DataFrames 的最佳做法是什么，我应该如何处理？我的伪代码会起作用，还是不建议在 for 循环中修改熊猫数据框？

标签： pythonpandas

解决方案

我认为最简单的是选择所有不以数字开头的列filterwith regex-^是字符串的开头，而\D不是数字：

df1 = df.filter(regex='^\D')

类似的选择：

df1 = df.loc[:, df.columns.str.contains('^\D')]

或逆条件并选择数字：

df1 = df.loc[:, ~df.columns.str.contains('^\d')]

df1 = df.loc[:, ~df.columns.str[0].str.isnumeric()]

如果想使用你的伪代码：

for col in df.columns:
    if col[0].isnumeric():
        df = df.drop(col, axis=1)

样品：

df = pd.DataFrame({'2A':list('abcdef'),
                   '1B':[4,5,4,5,5,4],
                   'C':[7,8,9,4,2,3],
                   'D3':[1,3,5,7,1,0],
                   'E':[5,3,6,9,2,4],
                   'F':list('aaabbb')})

print (df)
   1B 2A  C  D3  E  F
0   4  a  7   1  5  a
1   5  b  8   3  3  a
2   4  c  9   5  6  a
3   5  d  4   7  9  b
4   5  e  2   1  2  b
5   4  f  3   0  4  b

df1 = df.filter(regex='^\D')
print (df1)
   C  D3  E  F
0  7   1  5  a
1  8   3  3  a
2  9   5  6  a
3  4   7  9  b
4  2   1  2  b
5  3   0  4  b

推荐阅读