python - Pandas 删除数据框中指定字符后的部分字符串

Question

我想要一个简单的方法来删除数据框中指定字符后的字符串部分。这是一个简化的示例:

df:

   obs         a  b  c  d
0    1   1-23-12  1  2  3
1    2  12-23-13  4  5  5
2    3  21-23-14  4  5  5

我想删除 a 列中第一个 - 符号后的部分，我的预期输出是:

新的:

   obs   a  b  c  d
0    1   1  1  2  3
1    2  12  4  5  5
2    3  21  4  5  5

Answer 1

您可以通过将重新格式化函数传递给 apply 方法来重新格式化这些值，如下所示:

from StringIO import StringIO
import pandas as pd

data = """   obs  a  b  c  d
1   1-23-12  1  2  3
2  12-23-13  4  5  5
3  21-23-14  4  5  5"""

# Build dataframe from data
df = pd.read_table(StringIO(data), sep='  ')

# Reformat values for column a using an unnamed lambda function
df['a'] = df['a'].apply(lambda x: x.split('-')[0])

这会给你想要的结果:

   obs   a  b  c  d
0    1   1  1  2  3
1    2  12  4  5  5
2    3  21  4  5  5