karpanGit · July 27, 2025 14:10
diff --git a/Pandas assign with lambda function with additional (default) arguments.py b/Pandas assign with lambda function with additional (default) arguments.py
 # simple dataframe with three columns
 df = pd.DataFrame({'a': [1, 2, 3, 4, 5],
                   'b': [5, 4, 3, 2, 1],
                   'c': ['A', 'B', 'C', 'D', 'E']})

 df.assign(**{col: lambda df_, col=col: df[col]*2 for col in ['a', 'b']})

 #     a   b  c
 # 0   2  10  A
 # 1   4   8  B
 # 2   6   6  C
 # 3   8   4  D
 # 4  10   2  E
	# simple dataframe with three columns
	df = pd.DataFrame({'a': [1, 2, 3, 4, 5],
	'b': [5, 4, 3, 2, 1],
	'c': ['A', 'B', 'C', 'D', 'E']})

	df.assign(*{col: lambda df_, col=col: df[col]2 for col in ['a', 'b']})

	# a b c
	# 0 2 10 A
	# 1 4 8 B
	# 2 6 6 C
	# 3 8 4 D
	# 4 10 2 E
No results found