gấu trúc chuyển đổi cột của tôi từ số nguyên sang phao.

Question

shift chuyển đổi cột của tôi từ số nguyên sang phao. Nó chỉ ra rằng np.nan là chỉ nổi. Có cách nào để giữ cột chuyển đổi dưới dạng số nguyên không?

df = pd.DataFrame({"a":range(5)}) 
df['b'] = df['a'].shift(1) 

df['a'] 
# 0 0 
# 1 1 
# 2 2 
# 3 3 
# 4 4 
# Name: a, dtype: int64 

df['b'] 

# 0 NaN 
# 1  0 
# 2  1 
# 3  2 
# 4  3 
# Name: b, dtype: float64 

Answer 1

Vấn đề là bạn sẽ có được NaN giá trị là những gì float, vì vậy int được chuyển thành float - xem na type promotions.

Một giải pháp khả thi là chuyển đổi NaN giá trị cho một số giá trị như 0 và sau đó có thể chuyển đổi để int:

df = pd.DataFrame({"a":range(5)}) 
df['b'] = df['a'].shift(1).fillna(0).astype(int) 
print (df) 
    a b 
0 0 0 
1 1 0 
2 2 1 
3 3 2 
4 4 3 

Answer 2

Bạn có thể xây dựng một mảng numpy bởi prepending một 0 cho tất cả nhưng yếu tố cuối cùng của cột a

df.assign(b=np.append(0, df.a.values[:-1])) 

    a b 
0 0 0 
1 1 0 
2 2 1 
3 3 2 
4 4 3 

Answer 3

một giải pháp khác là sử dụng hàm replace() và nhập loại

df['b'] = df['a'].shift(1).replace(np.NaN,0).astype(int)