pandas去掉空格|极客笔记

pandas去掉空格

在数据处理过程中，经常会遇到字符串中包含空格的情况，这些空格可能会影响后续的数据分析和处理。在使用pandas进行数据处理时，我们经常需要去除字符串中的空格。本文将详细介绍如何使用pandas去掉字符串中的空格。

1. pandas中去掉字符串中的空格

在pandas中，可以使用str.strip()方法去除字符串两端的空格，使用str.lstrip()方法去除字符串左端的空格，使用str.rstrip()方法去除字符串右端的空格。

首先我们创建一个包含空格的示例数据：

import pandas as pd

data = {'A': ['  apple ', 'banana  ', '  orange  ', 'peach', 'melon']}
df = pd.DataFrame(data)
print(df)

运行以上代码，得到的示例数据如下：

             A
0     apple 
1    banana  
2     orange  
3      peach
4      melon

接下来，我们分别使用str.strip()、str.lstrip()和str.rstrip()去除字符串中的空格：

df['A_stripped'] = df['A'].str.strip()
df['A_lstripped'] = df['A'].str.lstrip()
df['A_rstripped'] = df['A'].str.rstrip()

print(df)

运行以上代码，得到去除空格后的数据如下：

             A   A_stripped  A_lstripped  A_rstripped
0     apple          apple       apple        apple
1    banana        banana      banana       banana
2     orange         orange        orange       orange
3      peach         peach        peach        peach
4      melon         melon       melon        melon

通过以上示例可以看出，空格已经被成功去除。

2. 去除指定列中的空格

如果我们只需要对某一列数据进行空格去除操作，可以直接对该列进行操作：

df['A'] = df['A'].str.strip()

print(df)

运行以上代码，查看去除空格后的数据：

             A   A_stripped  A_lstripped  A_rstripped
0     apple          apple       apple        apple
1    banana        banana      banana       banana
2     orange         orange        orange       orange
3      peach         peach        peach        peach
4      melon         melon       melon        melon

3. 去除所有列中的空格

如果我们要对所有列中的空格进行去除，可以使用applymap()方法：

df = df.applymap(lambda x: x.strip() if isinstance(x, str) else x)

print(df)

运行以上代码，去除所有列中的空格后的数据如下：

        A  A_stripped  A_lstripped  A_rstripped
0  apple       apple       apple        apple
1  banana     banana      banana       banana
2  orange     orange        orange       orange
3  peach       peach        peach        peach
4  melon       melon       melon        melon

通过以上操作，我们成功去除了所有列中的空格。