DataFrame对象 .assign() 函数的参数
时间: 2024-04-29 20:25:06 浏览: 60
对象作为函数参数
4星 · 用户满意度95%
.assign()函数的参数是一个或多个关键字参数,每个关键字参数对应一个新列的名称和取值的表达式。例如:
```python
df_new = df.assign(new_column_1 = expression_1, new_column_2 = expression_2, ...)
```
其中,`new_column_1`、`new_column_2`等为新列的名称,`expression_1`、`expression_2`等为用于计算新列取值的表达式。
这些表达式可以是常数值、变量、函数、条件语句等任何合法的Python表达式。在表达式中,可以使用当前DataFrame对象的列名作为变量,例如:
```python
df_new = df.assign(new_column_1 = df['column_1'] + df['column_2'], new_column_2 = np.log(df['column_3']))
```
这个例子中,新列`new_column_1`的取值为`column_1`和`column_2`两列对应元素的和,新列`new_column_2`的取值为`column_3`列元素的自然对数。
阅读全文