要将多个变量传递给普通的python函数,您可以编写如下内容:
def a_function(date,string,float):
do something....
convert string to int,
date = date + (float * int) days
return date
使用Pandas数据帧时,我知道您可以根据以下内容创建新列:
df['new_col']) = df['column_A'].map(a_function)
For example this might return the year from a date column
return date.year
我想知道的是,您可以将多个数据传递给单个函数(如上面的第一个示例所示),您是否可以在创建新的pandas dataframe列时使用多个列?
例如,将日期Y-M-D的三个独立部分组合成一个字段 .
df['whole_date']) = df['Year','Month','Day'].map(a_function)
我通过以下测试得到了一个关键错误 .
def combine(one,two,three):
return one + two + three
df = pd.DataFrame({'a': [1,2,3], 'b': [2,3,4],'c': [4,5,6]})
df['d'] = df['a','b','b'].map(combine)
有没有办法使用.MAP或其他东西在pandas数据框中创建一个新列,它将三列作为输入并返回一列 . 例如,输入将是1,2,3,输出将是1 * 2 * 3
同样,还有一种方法是让函数接受一个参数,一个日期并返回三个新的pandas dataframe列;年,月,日一个?
2 回答
为此,您可以将
apply
与axis=1
一起使用 . 但是,不是使用三个单独的参数(每列一个)调用,而是使用每个行的单个参数调用指定的函数,该参数将是包含该行数据的Series . 你可以在你的功能中考虑到这一点:或者你可以传递一个lambda,它将Series解包为单独的参数:
如果只想传递特定的行,则需要通过使用列表在DataFrame上 Build 索引来选择它们:
请注意双括号 . (这与
apply
没有任何关系;使用列表索引是从DataFrame访问多个列的常规方法 . )但是,它需要使用
apply
,因为您可以在列本身上使用向量化操作 . 上面的combine
函数可以简单地使用DataFrame列本身作为参数调用:当“组合”操作是可矢量化时,这通常更有效 .
如上所述,有两种基本方法可以做到这一点:使用
apply
的一般但非矢量化方式,以及更快的矢量化方式 . 假设你有一个像这样的DataFrame:您可以定义一个函数,为每个值返回一个Series,然后
apply
为该列:在这种情况下,这是唯一的选择,因为没有矢量化方式来访问各个日期组件 . 但是,在某些情况下,您可以使用矢量化操作:
这里再次通过直接操作值而不是元素地应用函数来对操作进行矢量化 . 在这种情况下,我们有两个向量化操作(获取第一个字符并获取字符串长度),然后我们将结果包装在另一个DataFrame调用中,为两种结果中的每一种创建单独的列 .
我通常使用
apply
来做这种事情;它基本上是map的DataFrame版本(axis参数允许您决定是将函数应用于行还是列):要么