我怎么做像numpy where(2darray ==0，1darray，1darray)这样的东西？

import numpy as np
import pandas as pd

def custom_where(condition, x, y):
    result = []
    for i in range(len(condition)):
        row = []
        for j in range(len(condition[i])):
            if condition[i][j] == 0:
                row.append(x[j])
            else:
                row.append(y[j])
        result.append(row)
    return np.array(result)

# 示例数据
condition = np.array([[0, 1, 0], [1, 0, 1]])
x = np.array([1, 2, 3])
y = np.array([4, 5, 6])

# 调用自定义的where函数
output = custom_where(condition, x, y)

# 打印结果
print(output)