如何找到一个字母表并在 Pandas 中提取字母表和与之一起标记的数字？ [复制]

Question

我想在数据框中创建一个新列，用于在列中搜索字母表。基于它，它将搜索下一个数字并将字母和数字复制到新提取的列中。例子：

Month	Sem_Year
2020-04-01	H1 2020
2020-05-01	2020 H1
2020-06-01	H1 2020
2020-07-01	H2 2020
2020-08-01	H2 2020
2020-09-01	2020 H2
2020-10-01	2020 H2
2020-11-01	H2 2020
2020-12-01	H2 2020
2021-01-01	H1 2021
2021-02-01	H1 2021

现在我想在第二列中搜索字母 H 并提取与其一起标记的字母和数字。例子：

Month	Sem_Year	Sem
2020-04-01	H1 2020	H1
2020-05-01	2020 H1	H1
2020-06-01	H1 2020	H1
2020-07-01	H2 2020	H2
2020-08-01	H2 2020	H2
2020-09-01	2020 H2	H2
2020-10-01	2020 H2	H2
2020-11-01	H2 2020	H2
2020-12-01	H2 2020	H2
2021-01-01	H1 2021	H1
2021-02-01	H1 2021	H1

Answer 1

您可以使用 df.insert() 添加新列。要提取字母表，循环遍历第二列中的值 (column_value) 并使用“value_for_new_column=column_value.split(' ')[0]”

Answer 2

假设您只需要新列的前两个字符，那么您可以使用：

df['Sem'] = df['Sem_year'].str[0:2]

相同的原理可用于提取更复杂的子字符串，例如使用正则表达式

或者，如果您想拆分字符串中的空格，则可以使用：

df['Sem'] = df['Sem_year'].str.split().str[0]

对于您提供的数据样本，两者都给出相同的结果。