如何在Python(使用Itertools)中生成所有不同的组合(其中输入元素重复)？

from itertools import combinations_with_replacement

def generate_combinations(elements):
    combinations = []
    for r in range(1, len(elements) + 1):
        combinations.extend(combinations_with_replacement(elements, r))
    return combinations

# 输入元素
elements = ['A', 'B', 'C']

# 生成所有不同的组合
result = generate_combinations(elements)

# 打印结果
for combination in result:
    print(combination)

('A',)
('B',)
('C',)
('A', 'A')
('A', 'B')
('A', 'C')
('B', 'B')
('B', 'C')
('C', 'C')