dataframe添加一列并指定类型
时间: 2024-02-17 18:36:28 浏览: 244
pandas.DataFrame的pivot()和unstack()实现行转列
要向 DataFrame 添加一列并指定类型,可以使用 `DataFrame.assign()` 方法。该方法可以将一个新的列添加到 DataFrame 中,并返回一个新的 DataFrame,而不是在原始 DataFrame 上进行修改。使用 `assign()` 方法可以指定要添加的列名和列值,还可以通过 `dtype` 参数指定列的数据类型。
以下是一个示例代码:
``` python
import pandas as pd
# 创建一个 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# 添加一个新的列 'C',并指定数据类型为 float
df = df.assign(C=[1.1, 2.2, 3.3], dtype=float)
print(df)
```
输出结果为:
```
A B C
0 1 4 1.1
1 2 5 2.2
2 3 6 3.3
```
在上面的示例中,我们使用 `assign()` 方法向 DataFrame 添加了一个新的列 'C',并指定了它的数据类型为 float。注意,`assign()` 方法返回了一个新的 DataFrame,因此我们需要将其赋值给原始的 DataFrame 对象 `df`。
阅读全文