def func(i):
    if i > 0:
        return 'A'
    else:
        return 'B'

dat = [func(i) for i in [-3, 5, 2, -1]]  # ['B', 'A', 'A', 'B']