如何通过缩进在分割线中打印行?

问题描述 投票:0回答:2

我有字符串:

text = '''TextTextTextTextTextTextTextTextText1
        TextTextTextTextTextTextTextTextText1
    TextTextTextTextTextTextTextTextText2
        TextTextTextTextTextTextTextTextText2
        TextTextTextTextTextTextTextTextText2
        TextTextTextTextTextTextTextTextText2
        TextTextTextTextTextTextTextTextText2
    TextTextTextTextTextTextTextTextText3
        TextTextTextTextTextTextTextTextText3
        TextTextTextTextTextTextTextTextText3
    TextTextTextTextTextTextTextTextText4
        TextTextTextTextTextTextTextTextText4
        TextTextTextTextTextTextTextTextText4'''

我想通过缩进分割此字符串并将其添加到列表中。这是我当前的代码:

nr_lines = 0
indent_dict = {}
for line in summary1.splitlines(True):
    print(line)
    print("------------------------------")
    nr_lines+=1
    whitespaces_count = len(line) - len(line.lstrip())
    indent_dict[nr_lines] = whitespaces_count
print(indent_dict)

list_of_values = []

# Removed first key with value (indent) = 0
indent_dict_without = dict(indent_dict)
key = 1
del indent_dict_without[key]

# Adding values from dict to list
for key, value in indent_dict_without.items():
    list_of_values.append(value)
print(list_of_values)

# Finding minimum value
x = min(list_of_values)

list_of_small = []

for nr in list_of_values:
    if nr == x:
        list_of_small.append(nr)

print(list_of_small)

# Finding which line have all smallest indent
n = 0
key_1 = []
for key, value in indent_dict.items():
    if value == list_of_small[n]:
        key_1.append(key)
print(key_1)

输出为:

{1: 0, 2: 12, 3: 8, 4: 12, 5: 12, 6: 12, 7: 12, 8: 8, 9: 12, 10: 12, 11: 8, 12: 12, 13: 12} # dict with line and value (indent)
[12, 8, 12, 12, 12, 12, 8, 12, 12, 8, 12, 12] # list with indents
[8, 8, 8] # the smallest indents
[3, 8, 11] # lines for smallest indents

现在,我不知道如何拆分和添加这4个部分作为列表的元素:

list = ['TextTextTextTextTextTextTextTextText1
            TextTextTextTextTextTextTextTextText1',
        'TextTextTextTextTextTextTextTextText2
            TextTextTextTextTextTextTextTextText2
            TextTextTextTextTextTextTextTextText2
            TextTextTextTextTextTextTextTextText2
            TextTextTextTextTextTextTextTextText2',
        'TextTextTextTextTextTextTextTextText3
            TextTextTextTextTextTextTextTextText3
            TextTextTextTextTextTextTextTextText3',
        'TextTextTextTextTextTextTextTextText4
            TextTextTextTextTextTextTextTextText4
            TextTextTextTextTextTextTextTextText4']

我应该创建一个新变量并逐行添加行,直到新缩进吗?

python split indentation
2个回答
0
投票

这是我最快想到的。我敢肯定还有更优雅的解决方案

text = '''TextTextTextTextTextTextTextTextText1
    TextTextTextTextTextTextTextTextText1
TextTextTextTextTextTextTextTextText2
    TextTextTextTextTextTextTextTextText2
    TextTextTextTextTextTextTextTextText2
    TextTextTextTextTextTextTextTextText2
    TextTextTextTextTextTextTextTextText2
TextTextTextTextTextTextTextTextText3
    TextTextTextTextTextTextTextTextText3
    TextTextTextTextTextTextTextTextText3
TextTextTextTextTextTextTextTextText4
    TextTextTextTextTextTextTextTextText4
    TextTextTextTextTextTextTextTextText4'''


lines = text.split('\n')
# Count spaces in each line
indent_lst = [line.count(' ') for line in text.splitlines(True)]
# Find where indentation changes
indices = []
for idx in range(len(indent_lst[1:])): # Start at second element in list
    # Here I assume, that the indentation is constant. A change from more spaces to fewer spaces means,
    # that a new block has started
    if indent_lst[idx-1] > indent_lst[idx]: # Look back at previous element and compare with current
        indices.append(idx)

final_lst = []
# Use slicing to append from block to block
for idx in range(len(indices)):
    if indices.index(indices[idx]) == (len(indices) -1 ): # Take care of last block
        final_lst.append(''.join(lines[indices[idx]:]))
    else:
        final_lst.append(''.join(lines[indices[idx]:indices[idx+1]])) # Add block to final list
print(final_lst)

这里是结果:

['TextTextTextTextTextTextTextTextText1        TextTextTextTextTextTextTextTextText1', '    TextTextTextTextTextTextTextTextText2        TextTextTextTextTextTextTextTextText2        TextTextTextTextTextTextTextTextText2        TextTextTextTextTextTextTextTextText2        TextTextTextTextTextTextTextTextText2', '    TextTextTextTextTextTextTextTextText3        TextTextTextTextTextTextTextTextText3        TextTextTextTextTextTextTextTextText3', '    TextTextTextTextTextTextTextTextText4        TextTextTextTextTextTextTextTextText4        TextTextTextTextTextTextTextTextText4']

我希望这已经对您有所帮助,并随时询问您是否有问题!


0
投票

如果我对您的理解正确,您希望找到所有缩进最小的行。

我处理此问题的方法如下。我将创建一个defaultdict,将组成缩进的空格的数字作为键,并将包含该缩进的行的所有索引的列表作为值,作为一个值:

from collections import defaultdict

text = '''TextTextTextTextTextTextTextTextText1
        TextTextTextTextTextTextTextTextText1
    TextTextTextTextTextTextTextTextText2
        TextTextTextTextTextTextTextTextText2
        TextTextTextTextTextTextTextTextText2
        TextTextTextTextTextTextTextTextText2
        TextTextTextTextTextTextTextTextText2
    TextTextTextTextTextTextTextTextText3
        TextTextTextTextTextTextTextTextText3
        TextTextTextTextTextTextTextTextText3
    TextTextTextTextTextTextTextTextText4
        TextTextTextTextTextTextTextTextText4
        TextTextTextTextTextTextTextTextText4'''

def count_indentation(line):
    return len(line) - len(line.lstrip())

lines = text.splitlines(keepends=False)
indent_dict = defaultdict(list)
for idx, line in enumerate(lines):
    if count_indentation(line) > 0:
        indent_dict[count_indentation(line)].append(idx)

现在indent_dict看起来像:

defaultdict(list, {8: [1, 3, 4, 5, 6, 8, 9, 11, 12], 4: [2, 7, 10]})

接下来,我们用最小的键找到相关行的索引:

smallest_indent = min(indent_dict)
line_idexes_smalles_indents = indent_dict[smallest_indent]

line_idexes_smalles_indents的结果为[2, 7, 10]。现在,我们需要根据这些索引对原始文本进行分区。

def partition(lines, indices):
    return [''.join(lines[i:j]) for i, j in zip([0]+indices, indices+[None])]

partition(lines, line_idexes_smalles_indents)

结果:

['TextTextTextTextTextTextTextTextText1        TextTextTextTextTextTextTextTextText1',
 '    TextTextTextTextTextTextTextTextText2        TextTextTextTextTextTextTextTextText2        TextTextTextTextTextTextTextTextText2        TextTextTextTextTextTextTextTextText2        TextTextTextTextTextTextTextTextText2',
 '    TextTextTextTextTextTextTextTextText3        TextTextTextTextTextTextTextTextText3        TextTextTextTextTextTextTextTextText3',
 '    TextTextTextTextTextTextTextTextText4        TextTextTextTextTextTextTextTextText4        TextTextTextTextTextTextTextTextText4']
© www.soinside.com 2019 - 2024. All rights reserved.