Блог

Мои данные генерируются не совсем удобным способом: сначала несколько пробелов, затем номер индекса (1-12 в этом примере), а затем фактическое значение, связанное с индексом. Я хотел бы разделить строку на два списка: 1 список с индексами и 1 список со значениями. Я написал следующий код, который работает для того, что я хочу. Однако это кажется громоздким и занимает много секунд для запуска набора данных из нескольких тысяч строк. Есть ли способ ускорить это для больших наборов данных?

 data = ['         11.814772E3',
 '         2-1.06152E3',
 '         33.876477E1',
 '         4-2.65704E3',
 '         51.141537E4',
 '         61.378482E4',
 '         71.401565E4',
 '         86.782599E3',
 '         9-1.22921E3',
 '        103.400054E3',
 '        111.558086E3',
 '        121.017818E4']

values_total = [] #without empty strings
location     = [] #index when id goes to value
ids          = [] #Store ids
values       = [] #Store values

step_array = np.linspace(1,1E3,1E3) #needed to calculate index values

for i in range(len(data)):

    #Check how many indices have to be removed
    location.append([])
    location[i].append(int(math.log10(step_array[i])) 1)

    #Store values after empty strings
    for j in range(len(data[i])):
        values_total.append([])
        if data[i][j] != ' ':
            values_total[i].append(data[i][j])

    #Split list based on calculated lengths
    ids.append(values_total[i][:location[i][0]])
    values.append(values_total[i][location[i][0]:])
 

Вы можете попробовать использовать приведенный ниже код:

 indices = []
vals = []
for i, d in enumerate(data, 1):  # enumerate starting from 1, so we know current index
    tmp = d.strip()  # remove whitespace
    split_idx = len(str(i))  # figure out the length of the current index
    indices.append(i)  # current index
    vals.append(float(tmp[split_idx:]))  # everything after current index length