Python：基于索引生成字符串数组

Question

我有两个数组定义为：

Winter   =   np.array( [ 9, 10, 11, 12 ] ) 
Summer   =   np.array( [ 4, 5, 6, 7, 8 ] )
Other    =   np.array( [ 1, 2, 3 ] )

我有另一个月的数组：

GivenMonths = np.array([ 3, 4,  8,  8, 10,  7])

我想生成一个SeasonIdentifier：

SeasonIdentifier = [ 'Other', 'Summer', 'Summer', 'Summer', 'Winter', 'Summer' ] 

即。对于GivenMonths的每个元素，我想查找它是来自Winter还是Summer数组，并将该信息存储在SeasonIdentifier中。

编辑： 根据以下评论对我的原始问题进行了更改

Answer 1

这是你想要的吗？

SeasonIdentifier =  ['Summer' if x in Summer else 'Winter' if x in Winter else 'Other' for x in GivenMonths]