如何遍历Python列表获取特定累加序列元素并拼接为字符串？

我有一个名为data的Python列表，示例为data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, ...]，需要遍历该列表提取符合**1、3、6、10、15、21……**这类规律的元素，并将其拼接成类似data_string = "1, 3, 6, 10"格式的字符串。列表data的元素数量不固定且可能变化。

我尝试运行了以下函数：

def create_lists(values_list):
    result = [[]]
    for value in values_list[1:]:
        new_lists = [curr_list + [value] for curr_list in result]
        result.extend(new_lists)
    return result

result = create_lists(values_list)
print(result)

但该函数仅输出包含单个元素的独立子列表，不知道该怎么解决。

首先明确你要提取的数列规律：这是三角形数，第n项的计算公式为 $n(n+1)/2$（n从1开始）。比如第1项是1×2/2=1，第2项是2×3/2=3，第3项3×4/2=6，以此类推。

接下来分两步处理：

1. 遍历data列表，筛选出属于三角形数的元素；
2. 将筛选后的元素拼接成指定格式的字符串。

具体实现代码 #

def is_triangular_number(num):
    # 通过数学推导判断是否为三角形数：解方程n² +n -2num=0，判别式需为完全平方数，且正根为整数
    discriminant = 1 + 8 * num
    sqrt_disc = int(discriminant ** 0.5)
    return sqrt_disc * sqrt_disc == discriminant and (sqrt_disc - 1) % 2 == 0

def get_triangular_string(data):
    # 筛选符合条件的元素并转为字符串
    triangular_elements = [str(num) for num in data if is_triangular_number(num)]
    # 拼接成目标格式
    return ", ".join(triangular_elements)

# 示例使用
data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
data_string = get_triangular_string(data)
print(data_string)  # 输出：1, 3, 6, 10, 15

代码说明 #

• is_triangular_number函数：利用三角形数的数学性质做判断，避免了生成大量三角形数去匹配的低效操作。
• get_triangular_string函数：先完成筛选，再通过join方法高效拼接字符串，符合需求格式。

原函数问题说明 #

你写的create_lists函数实际是在生成输入列表的所有子集，和提取特定规律元素的需求完全不匹配，所以才会输出一堆无关的子列表。

内容的提问来源于stack exchange，提问作者Charles Jr