python - 如何在 Pandas 中获取浮点列的整数部分

Question

假设我有一个如下所示的数据框 df

    qty
0   1.300
1   1.909

现在我只想提取 qty 列的整数部分，df 应该看起来像

   qty
0   1
1   1

尝试使用 df['qty'].round(0) 但没有得到想要的结果，因为它将数字四舍五入到最接近的整数。

Java 有一个函数 intValue() 可以执行所需的操作。 pandas有类似的功能吗？

Answer 1

通过 Series.astype 将值转换为整数:

df['qty'] = df['qty'].astype(int)
print (df)
   qty
0    1
1    1

如果上述方法不起作用，可以使用 numpy.modf在 . 之前提取值:

a, b = np.modf(df['qty'])
df['qty'] = b.astype(int)
print (df)
   qty
0    1
1    1

或者在 . 之前拆分，但如果 DataFrame 很大，它应该很慢:

df['qty'] = b.astype(str).str.strip('.').str[0].astype(int)

或者使用numpy.floor :

df['qty'] = np.floor(df['qty']).astype(int)