def METHOD_NAME(psi):
    # ps2xyz([1, 2, 2, 0]) = {"x": [0], "y": [1, 2], "z": []}
    zs = []  # no x or y for QUBO problem
    for i, j in enumerate(psi):
        if j == 3:
            zs.append(i)
    return zs