在Pandas数据框架中分割一列并获得其中的一部分

  • Post category:Python

在 Pandas 数据框架中,分割一列并获取其中的一部分可通过字符串方法实现。

具体步骤如下:

  1. 读取数据并创建数据框架
import pandas as pd

# 读取数据
data = pd.read_csv('data.csv')

# 创建数据框架
df = pd.DataFrame(data)
  1. 分割一列并获取其中的一部分

假设我们要获取 Name 这一列数据的第一个单词。

# 获取第一个单词
df['First name'] = df['Name'].str.split(' ').str.get(0)

这里使用了字符串方法 split ,将 Name 列按照空格分割成多个部分,再使用 get 方法获取第一个部分,最后将获取到的结果存入一个新列 First name 中。

  1. 查看结果
print(df.head())

输出结果如下:

   PassengerId                                               Name     Sex  ...  Embarked  First name
0            1                            Braund, Mr. Owen Harris    male  ...         S     Braund,
1            2  Cumings, Mrs. John Bradley (Florence Briggs Th...  female  ...         C    Cumings,
2            3                             Heikkinen, Miss. Laina  female  ...         S  Heikkinen,
3            4       Futrelle, Mrs. Jacques Heath (Lily May Peel)  female  ...         S   Futrelle,
4            5                           Allen, Mr. William Henry    male  ...         S      Allen,

可以看到,新的一列 First name 已经被添加到了数据框架中,并包含了每个乘客的第一个单词。

这就是在 Pandas 数据框架中分割一列并获取其中的一部分的完整攻略。