将定义的函数应用于Python上的多行

Question

我正在尝试定义一个函数，以便一次适用于多行。

这是我想做的。

我有A列，其中包含有关消费者渠道设置的付款的信息。我正在尝试将其转换为单独的二进制列（标志）。对于如果列a = source1，那么我想创建一个名为source1的列，该列将填充1，否则0。对于a = source2，然后列名source2将填充1，else 0。等等。 以下是我的代码的示例：

def payer(row1,row2,row3):
if file['a'] == 'Source1':
    return {row1:1, row2:0, row3:0}
elif file['a'] == 'Source2':
    return {row1:0, row2:1, row3:0}
elif file['a'] == 'Sourc3':
    return {row1:0, row2:0, row3:1}
else:
   return {row1:0, row2:0, row3:0}
file[['Source1','Source2','Source3']] =""
file[['Source1','Source2','Source3']]= file[['Source1','Source2','Source3']].apply(lambda a:(a), axis=0)

我绝对是新的，但是一直在搜索一段时间，但找不到任何解决方案。请帮忙！

---

Answer 1

import pandas as pd

def payer(name):
    if name == 'IB Plans':
        return 1, 2, 3
    elif name == 'Web Plans':
        return 1, 1, 3
    elif name == 'NC Payers':
        return 1, 1, 3
    else:
        return 2, 2, 2

file = pd.DataFrame({'a': ['IB Plans', 'Web Plans', 'NC Payers', 'Some other', 'NC Payers']})
file[['x', 'y', 'z']] = file.apply(lambda row: payer(row['a']), axis=1, result_type='expand')
print(file)

输出

            a  x  y  z
0    IB Plans  1  2  3
1   Web Plans  1  1  3
2   NC Payers  1  1  3
3  Some other  2  2  2
4   NC Payers  1  1  3