如果 NaN 在里面，则 Pandas 数字列将视为字符串

Question

我开始学习 Python，但我对 pandas 数据框有疑问。在 R 中，即使数字列具有 NaN 值，R 也设法在每一列中定义正确的数据类型。在熊猫中，情况似乎并非如此：

data = {
"calories": ["NA", 380, 390],
"duration": [50, 40, 45]
}

df = pd.DataFrame(data)
df.dtypes

如何设法自动检测每列中正确的数据类型？

提前致谢

Answer 1

“NA”是一个字符串，使用np.nan或float('nan')，或pd.NA：

data = {
"calories": [float('nan'), 380, 390],
"duration": [50, 40, 45]
}

df = pd.DataFrame(data)
print(df.dtypes)

calories    float64
duration      int64
dtype: object