python-list-comprehension-three-way-partitioning-array-around-given-range
Python 列表理解|给定范围内数组的三向分区
原文:https://www . geesforgeks . org/python-list-understance-三路-分区-数组-围绕给定范围/
给定一个数组和一个范围**【low val，high val】，围绕该范围对数组进行划分，使数组分为三部分。
1)所有比小的元素都是第一位的。
2)范围内的所有元素低值到高值接下来。
3)所有大于高值的元素出现在最后。
三套单个元素可以任意顺序出现。**
示例:
Input: arr = [1, 14, 5, 20, 4, 2, 54, 20, 87, 98, 3, 1, 32]
lowVal = 14, highVal = 20
Output: arr = [1, 5, 4, 2, 3, 1, 14, 20, 20, 54, 87, 98, 32]
Input: arr = [1, 14, 5, 20, 4, 2, 54, 20, 87, 98, 3, 1, 32]
lowVal = 20, highVal = 20
Output: arr = [1, 14, 5, 4, 2, 3, 1, 20, 20, 54, 87, 98, 32]
这个问题我们已经有了解决方案，请参考给定范围内数组的三向分区链接。我们可以使用列表理解在 python 中快速解决这个问题。方法很简单，
- 将列表分为三部分，第一部分包含小于 lowVal 的元素，第二部分包含介于 lowVal 和 highVal 之间的元素，第三部分包含大于 highVal 的元素。
- 将所有三个部分连接在一起。
# Function for Three way partitioning of an
# array around a given range
def threeWay(input, lowVal, highVal):
# separate input list in three parts
first = [ num for num in input if num<lowVal ]
second = [ num for num in input if (num>=lowVal and num<=highVal) ]
third = [ num for num in input if num>highVal ]
# concatenate all three parts
print(first + second + third)
# Driver program
if __name__ == "__main__":
input = [1, 14, 5, 20, 4, 2, 54, 20, 87, 98, 3, 1, 32]
lowVal = 14
highVal = 20
threeWay(input, lowVal, highVal)
输出:
[1, 5, 4, 2, 3, 1, 14, 20, 20, 54, 87, 98, 32]