Python是一門常用於數據處理的編程語言，而對於數據處理來說，取最後一列也是常用的操作之一。本文將介紹Python取最後一列的方法。
一、split方法
對於文本文件或者字符串，我們可以使用split方法對每行進行切分，並取出最後一列。split方法將行按照給定字符進行分割，返回的是一個列表，我們只需要取最後一個元素即可。
with open('data.txt', 'r') as f:
for line in f:
cols = line.strip().split(',')
last_col = cols[-1]
print(last_col)
上述代碼中使用with open語句，自動關閉文件，避免了手動close帶來的繁瑣。strip方法用來去除行末尾的空白符，split方法按照’,’進行分割並返回一個列表，使用列表索引-1取出最後一個元素。
二、pandas庫
如果我們的數據存儲在pandas的DataFrame中，我們可以用loc和iloc方法直接取出最後一列。
import pandas as pd
df = pd.read_csv('data.csv')
last_col = df.iloc[:, -1]
print(last_col)
iloc方法通過位置索引選取行和列，這裡我們選取所有行（用冒號表示），最後一列則通過-1索引選取。
三、numpy庫
類似於pandas，如果我們的數據存儲在numpy的array中，我們可以用切片取出最後一列。
import numpy as np
arr = np.genfromtxt('data.csv', delimiter=',')
last_col = arr[:, -1]
print(last_col)
genfromtxt方法讀取文件，delimiter參數指定列分隔符。array的切片用冒號表示所有行，-1索引則表示最後一列。
四、總結
本文介紹了三種Python取最後一列的方法，分別是split方法、pandas庫和numpy庫。對於文本文件或者字符串，我們可以使用split方法進行操作。對於數據存儲在pandas的DataFrame或者numpy的array中，我們可以直接使用loc、iloc和切片等方法進行操作。以上三種方法各有優缺點，可以根據實際應用場景選擇使用。
原創文章，作者：CQAVB，如若轉載，請註明出處：https://www.506064.com/zh-hant/n/373541.html
微信掃一掃 
支付寶掃一掃 