python - 使用惯用的 Python 删除 pandas 列中的空格和换行符？

Question

我正在使用以下方法替换 pandas 数据框列标题中的所有空格和换行符。

我的问题是:

在下面的代码中使用列表理解是一种更有效的循环方式吗？

def headerfiller(df):
    for i in [" ","\n"]:
        df.columns = [c.replace(i,"_") for c in df.columns]

Answer 1

您可以使用可用于索引对象的字符串方法，在本例中为 columns.str.replace()，它允许您在不循环遍历值的情况下执行此操作:

In [23]: df = pd.DataFrame(np.random.randn(3,3), columns=['a\nb', 'c d', 'e\n f'])

In [24]: df.columns
Out[24]: Index([u'a\nb', u'c d', u'e\n f'], dtype='object')

In [25]: df.columns.str.replace(' |\n', '_')
Out[25]: Index([u'a_b', u'c_d', u'e__f'], dtype='object')

并且通过使用正则表达式，您可以同时替换空格和换行符。请参阅文档:http://pandas.pydata.org/pandas-docs/stable/generated/pandas.Series.str.replace.html (Series，Index方法相同)