Назначение списка массиву в python

Question

У меня есть файл данных «list_2_array.dat», как показано ниже. Во-первых, я хочу прочитать его, а затем я хочу взять под контроль элементы четвертого столбца для дальнейших математических операций.

1 2 3 10 
4 5 6 20 
1 3 5 30 
2 1 4 40 
3 2 3 50 

Я попытался следующий фрагмент кода

b_list = [] 
file=open('/path_to_file/list_2_array.dat', 'r') 
m1=[(i.strip()) for i in file] 

for j in m1: 
    b_list.append(j.replace('\n','').split(' ')) 

for i in range(5): 
    print b_list[i][3] 

, который дает OUTPUT

10 
20 
30 
40 
50 

Я не хочу, чтобы напечатать элементы, я заинтересован в первом присвоении четвертого элементов столбцов в 1-мерный массив, чтобы я мог легко обработать их позже. Я попробовал несколько способов сделать это, как показано ниже, но не работает

import numpy as np 
    for i in range(5): 
     arr = array (b_list[i][3]) 

Answer 1

f=open('/path_to_file/list_2_array.dat', 'r') 
l = [] 
for line in f.readlines(): 
    l.append(int(line.strip().split()[-1])) 
array=np.array(l) 

или более вещий я думаю ..:

f=open('/path_to_file/list_2_array.dat', 'r') 
l = [int(line.strip().split()[-1]) for line in f.readlines()] 
array=np.array(l) 

Answer 2

data = """1 2 3 10 
4 5 6 20 
1 3 5 30 
2 1 4 40 
3 2 3 50""" 
fourth = [int(line.split()[3]) for line in data.split("\n")] 
print(fourth) 

Выход:

[10, 20, 30, 40, 50] 

Answer 3

def get_last_col(file): 
last_col = [int(line.split()[-1]) for line in open(file)] 
return last_col 

Answer 4

Прежде всего, никогда не назначайте имена переменных, такие как str, file, int. рядом вы были почти там.

b_list = [] 
c_list = [] 
file=open('/path_to_file/list_2_array.dat', 'r') 
m1=[(i.strip()) for i in file] 

for j in m1: 
    b_list.append(j.replace('\n','').split(' ')) 

for i in range(5): 
    c_list.append(b_list[i][3]) 
print c_list 

Я не очень нравится это решение, поэтому я изменил @ user2994666 его/ее решение:

file_location = "/path_to_file/list_2_array.dat" 
def get_last_col(file_location): 
    last_col = [int(line.split()[-1]) for line in open(file_location)] 
    return last_col 

print get_last_col(file_location) 

Обратите внимание, что [-1] решение дает последний столбец, в вашем случае это дает без проблем. Если у вас есть файл с 5 столбцами, и вы по-прежнему интересуетесь 4-м, вы используете [3] вместо [-1]