溫馨提示×

Python中assign函數(shù)怎么使用

小億
266
2023-10-09 15:58:31
欄目: 編程語言

在Python中，assign函數(shù)不是Python內(nèi)置函數(shù)，而是pandas庫中的一個(gè)方法。assign方法用于在DataFrame中添加新的列或修改已有列的值。

assign方法的用法如下：

df.assign(**kwargs)

其中，df是一個(gè)DataFrame對(duì)象，kwargs是一個(gè)或多個(gè)關(guān)鍵字參數(shù)，每個(gè)關(guān)鍵字參數(shù)是一個(gè)字典，字典的鍵是要?jiǎng)?chuàng)建或修改的列名，字典的值是要為該列設(shè)置的值。

示例：

import pandas as pd
# 創(chuàng)建一個(gè)DataFrame
df = pd.DataFrame({'A': [1, 2, 3]})
# 使用assign方法添加新的列
df = df.assign(B=[4, 5, 6])
print(df)

輸出結(jié)果：

A  B
0  1  4
1  2  5
2  3  6

在上述示例中，使用assign方法向DataFrame中添加了一個(gè)名為B的新列，并設(shè)置了該列的值。

除了可以添加新列，assign方法還可以用于修改已有列的值。例如：

import pandas as pd
# 創(chuàng)建一個(gè)DataFrame
df = pd.DataFrame({'A': [1, 2, 3]})
# 使用assign方法修改已有列的值
df = df.assign(A=df['A'] * 2)
print(df)

輸出結(jié)果：

A
0  2
1  4
2  6

在上述示例中，使用assign方法修改了已有列A的值，將其乘以2。

0 踩