在Pandas中使用Apply结合含多条件判断的Lambda函数

It sounds like you’re hitting a common snag with pandas lambda functions: regular if/else statement syntax doesn’t play nice here because lambdas are restricted to a single expression, not multi-line statements. Let’s walk through the correct way to handle multiple conditions, plus a cleaner alternative for more complex categorization tasks.

Correct Lambda Syntax for Multiple Conditions #

Instead of writing separate if statements, you need to chain ternary conditional expressions in your lambda. Here’s how that looks with apply():

Suppose your DataFrame has a 'size' column, and you want to map values to categories like this:

• 'Small' if value < 10
• 'Medium' if 10 ≤ value < 20
• 'Large' if value ≥ 20

import pandas as pd

# Sample DataFrame
df = pd.DataFrame({'size': [7, 14, 22, 9, 19]})

# Chained ternary in lambda
df['category'] = df['size'].apply(
    lambda x: 'Small' if x < 10 else 'Medium' if 10 <= x < 20 else 'Large'
)

This works because each ternary is a single expression that evaluates directly to one of your category labels.

A Cleaner Alternative for Complex Logic #

If you have more than 2-3 conditions, chained ternaries can get messy and hard to debug. For these cases, numpy.select() is a better option—it lets you define conditions and their corresponding choices explicitly:

import numpy as np

# Define conditions and matching categories
conditions = [
    df['size'] < 10,
    (df['size'] >= 10) & (df['size'] < 20),
    df['size'] >= 20
]

choices = ['Small', 'Medium', 'Large']

# Apply the logic
df['category'] = np.select(conditions, choices, default='Unknown')

This is way easier to maintain, especially as your list of categories or conditions grows.

Why Your "Pass" Approach Might Have Failed #

If you tried using pass inside a lambda, that’s invalid because pass is a statement, not an expression. Lambdas can only return the result of a single expression—they can’t execute standalone statements like pass or multi-line if blocks. The methods above avoid this issue entirely by sticking to valid expressions.

内容的提问来源于stack exchange，提问作者aabujamra