Pandas ：从现有列中创建一个以字符结尾的新列

Question

这是图像

我需要选择多个以 ['edit'] 结尾的字符串，从这个数据框中创建新的列名 'state'。

Answer 1

这应该有效：

df['state'] = df.loc[df['YourCol'].str.endswith('[edit]'), 'YourCol']

只需更换YourCol与您正在使用的列， df与您的姓名dataframe 。 请注意，在不以[edit]结尾的地方，您将获得NaN值。

Answer 2

我认为您试图将“[edit]”行不仅仅是作为一个单独的列使用，而是将它们映射到它们下面的学校，这样您最终会得到这样的结果：

School                                           State

Fairbanks (University of Alaska Fairbanks)[2]    Alaska
Flagstaff (Northern Arizona University)          Arizona
Tempe (Arizona State University)                 Arizona
Tucson (University of Arizona)                   Arizona
Arkadelphia (Henderson State University, ...)    Arkansas

如果是这种情况，您可以将 Nicolas Gervais 的答案与 pandas 的fillna ：

df['State'] = df[df['YourCol'].str.endswith('[edit]')]['YourCol']
df['State'] = df['State'].fillna(method='ffill') # Forward-fills the states
df = df.drop(df[df['YourCol'].str.endswith('[edit]')]) # Drop the "header" rows
df['State'] = df['State'].str.slice_replace(start=-6) # Remove "[edit]" from the end of the state names