如何用pandas写一个简单的函数？

Question

我是python和pandas的新手。我已经编码了一些指令，我想从一个数据帧重复到另一个数据帧。

我想知道是否可以通过编写函数来重复该过程。

目标是：

1. 用唯一值替换列中的所有值
2. 将此列从一种类型转换为另一种类型
3. 将另一列的值乘以该列的值

以下是现有代码：

1。替换“单位”列的值

df = pd.DataFrame(df)

df.unit = '1000' 

2。将“单位”列的值转换为int格式

df['unit'] = df['unit'].astype('int')

3。将另一列的值（“值”）乘以该值

df["value"] = df[["value"]].multiply(df["unit"], axis="index")

是否可以将其转换为以列id（“ unit”，“ values”）作为变量的函数？

谢谢！

Answer 1

您可以编写如下函数：

def your_transfromation(df,column1,column2):
    df = pd.DataFrame(df)
    df[column1] = ['1000' for i in range(len(df))]
    df[column1] = df[column1].astype('int')
    df[column2] = df[[column2]].multiply(df[column1], axis="index")
    return df

并命名为：

df = your_transfromation(df,'unit',"value")