📅  最后修改于: 2023-12-03 15:10:02.138000             🧑  作者: Mango
在编程中,有时候需要打印所有小于某个数字N的数字,但是又只需要打印最多2个唯一数字。这个需求在实际的开发中很常见,比如打印某个范围内的不同年龄段用户数量。
下面给出一个Python代码示例,它能够实现这个需求。该程序首先生成一个包含所有小于N的数字的列表,然后使用set()函数去重,最后判断去重后的数字个数是否不超过2,如果不超过2则全部打印出来,否则只打印前2个数字。
def print_numbers_less_than_n_with_2_distinct_elements(n):
numbers = list(range(n))
distinct_numbers = set(numbers)
if len(distinct_numbers) <= 2:
print(distinct_numbers)
else:
print(set(list(distinct_numbers)[:2]))
例子:当n等于10时,程序输出:
{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}
例子:当n等于5时,程序输出:
{0, 1, 2, 3, 4}
例子:当n等于2时,程序输出:
{0, 1}
此代码可适用于大部分Python版本。