获取datafram中列的每个元素的第一个字符

Question

我有一个叫做df的数据帧。它有一个名为“Spec Type”的列。利用熊猫

df['Spec Type']

0         NaN
1          A1
2         G7V
3       F7+K4
.         .
.         .
169       A0e

我只想获取每个条目的第一个字符，并将其作为df的一个新列，名为“Spec Type Index”。但是，以下代码给了我一个错误：

^{pr2}$

'float' object is not subscriptable

错误在i[0]部分。我用它来获取索引元素的第一个字符。我该怎么办？在

Answer 1

df.loc[df['Spec Type'].notnull(), 'Spec Type Index'] = df['Spec Type'].str[0]

如果可以的话，问题是NaN是一个float数据类型，因此不能使用str方法，像上面这样的掩码可以避免这种情况

^{pr2}$