Python中的CASE WHEN语句|极客笔记

Python中的CASE WHEN语句

在SQL中，我们经常会用到CASE WHEN语句来实现条件判断。在Python中，虽然没有专门的CASE WHEN语句，但是我们可以通过一些技巧来实现类似的功能。本文将介绍在Python中如何实现类似于SQL中的CASE WHEN语句的功能。

1. 使用字典实现CASE WHEN语句

在Python中，我们可以使用字典来实现类似于CASE WHEN语句的功能。我们可以将条件作为字典的键，对应的结果作为字典的值。然后根据条件来查找对应的结果。

def case_when(condition, mapping, default=None):
    return mapping.get(condition, default)

result = case_when('A', {'A': 'Apple', 'B': 'Banana', 'C': 'Cherry'}, 'Unknown')
print(result)  # Output: Apple

在上面的示例中，我们定义了一个case_when函数，接受三个参数：condition表示条件，mapping表示条件和结果的映射关系，default表示如果条件不在映射中时的默认返回值。我们调用case_when函数来模拟CASE WHEN语句，根据条件返回对应的结果。

2. 使用函数实现CASE WHEN语句

除了使用字典，我们还可以使用函数来实现类似于CASE WHEN语句的功能。我们可以定义一个函数，根据条件返回对应的结果。

def case_when_func(condition):
    if condition == 'A':
        return 'Apple'
    elif condition == 'B':
        return 'Banana'
    elif condition == 'C':
        return 'Cherry'
    else:
        return 'Unknown'

result = case_when_func('B')
print(result)  # Output: Banana

在上面的示例中，我们定义了一个case_when_func函数，接受一个参数condition，根据不同的条件来返回对应的结果。我们调用case_when_func函数来实现条件判断功能。

3. 使用numpy实现CASE WHEN语句

在Python中，我们还可以使用numpy库来实现类似于CASE WHEN语句的功能。numpy库中的select函数可以实现根据条件返回不同的结果。

import numpy as np

condition = np.array(['A', 'B', 'C'])
result = np.select([condition == 'A', condition == 'B', condition == 'C'], ['Apple', 'Banana', 'Cherry'], default='Unknown')
print(result)  # Output: ['Apple' 'Banana' 'Cherry']

在上面的示例中，我们首先创建一个numpy数组condition，然后使用np.select函数来根据条件返回对应的结果。我们传入三个参数：条件数组，条件判断表达式，结果数组。最后输出根据条件判断的结果。