字符串拆分循环通过Dataframe

Question

我正在尝试使用Python循环遍历数据帧列,其格式如下:

Town 1, AL, USA
Town 2, AL, USA
Town 3, AK, USA
Town 4, CA, USA
Town 5, DE, USA
Town 6, MI, USA

我一直在尝试split()使用原始数据框(包括犯罪描述和URL列)和自己的列来使用该方法,作为数据框和Series对象.这些对象都没有可用的方法split().

所需的输出将是STATE缩写的另一列,所以我理解我正在尝试找到等效的a df.split(', ')并从该分割中追加系列或数据帧的第二个[1]索引.(如果我弄错了,请纠正我).

我该怎么做呢？

Answer 1

您可以使用矢量化字符串方法,例如df["col"].str.split(", ").str[1]:

>>> df
               col
0  Town 1, AL, USA
1  Town 2, AL, USA
2  Town 3, AK, USA
3  Town 4, CA, USA
4  Town 5, DE, USA
5  Town 6, MI, USA
>>> df["col"].str.split(", ")
0    [Town 1, AL, USA]
1    [Town 2, AL, USA]
2    [Town 3, AK, USA]
3    [Town 4, CA, USA]
4    [Town 5, DE, USA]
5    [Town 6, MI, USA]
Name: col, dtype: object
>>> df["col"].str.split(", ").str[1]
0    AL
1    AL
2    AK
3    CA
4    DE
5    MI
Name: col, dtype: object