Pandas：从列标题中提取前 4 位数字

Question

我有一个带有 +1000 列的熊猫数据框。 我只想从所有标题中提取前 4 位数字。

原始数据示例：

Q001-XYZ	Q002-XYZ
第一的	排
第二	排

我想得到：

Q001	Q002
第一的	排
第二	排

我试过 df.head() = df.head().str[:4] 但我收到错误 SyntaxError: can't assignment to function call

有什么建议吗？

Answer 1

列表理解是最快的选择：

df.columns = [col[:4] for col in df.columns]

Answer 2

尝试通过columns属性或通过rename()方法：

df.columns=df.columns.str.split('-').str[0]
#OR
df=df.rename(columns=lambda x:x.split('-')[0])

要么

df.columns=df.columns.str[:4]
#OR
df=df.rename(columns=lambda x:x[:4])

Answer 3

尝试使用regex ：

df.columns = df.columns.str.extract('([\w]+)-', expand=False)

Answer 4

你可以试试：

df.columns.str[:4].to_list()