String.split 停止在最后一個元素 python

Question

我正在使用 str.split 將正則表達式后的數據從一列拆分為兩列。

數據是：

DATA_SET 
1  98
2 176
3 277 
.... 
16 790

DATA_SET
1   8
2  37
3 378
.... 
16 899

我有一段代碼可用於單個數據集，將列拆分為兩列。

f = pd.read_table('my_file.txt')

f[['D1 ', 'D2']] = f['DATA_SET'].str.split(expand=True)

如何向代碼指定當它到達項目“16”時需要停止拆分操作？ 並在遇到下一個“DATA_SET”后再次執行

Answer 1

將拆分應用於您感興趣的行，即索引小於等於 16 的行

f[['D1 ', 'D2']] = f.loc[:16, 'DATA_SET'].\
    str.split(expand=True)