如何选择以＆＃34;持续时间＆＃34;开头的所有列或＆＃34;塑造＆＃34;？

Question

如何选择标题名称以＆＃34开头的所有列;持续时间＆＃34;或＆＃34;形状＆＃34;？ （而不是定义一长列列名）。我需要选择这些列并将空白字段替换为0。

column_names = ['durations.blockMinutes_x',
                'durations.scheduledBlockMinutes_y']
data[column_names] = data[column_names].fillna(0)

Answer 1

您可以使用数据框str的{​​{1}}方法：

startwith

或者您可以使用df = data[data.columns[data.columns.str.startwith('durations') | data.columns.str.startwith('so')]] df.fillna(0) 方法：

contains

Answer 2

使用my_dataframe.columns.values.tolist()获取列名称（基于Get list from pandas DataFrame column headers）：

column_names = [x for x in data.columns.values.tolist() if x.startswith("durations") or x.startswith("shape")]

Answer 3

我会使用select方法：

df.select(lambda c: c.startwith('durations') or c.startswith('shape'), axis=1)

Answer 4

一种简单的方法

data[data.filter(regex='durations|shape').columns].fillna(0)

示例屏幕截图