首頁(yè) 猿問(wèn) 只讀取特定行

只讀取特定行

Python

三國(guó)紛爭(zhēng) 2019-06-27 17:08:38

只讀取特定行我使用for循環(huán)來(lái)讀取文件，但是我只想讀取特定的行，比如第26行和第30行。有什么內(nèi)置的功能來(lái)實(shí)現(xiàn)這一點(diǎn)嗎？謝謝

查看完整描述

3 回答

皈依舞

TA貢獻(xiàn)1851條經(jīng)驗(yàn) 獲得超3個(gè)贊

如果要讀取的文件很大，并且不希望立即在內(nèi)存中讀取整個(gè)文件：

fp = open("file")for i, line in enumerate(fp):
    if i == 25:
        # 26th line
    elif i == 29:
        # 30th line
    elif i > 29:
        breakfp.close()

請(qǐng)注意i == n-1為n這條線。

在Python2.6或更高版本中：

with open("file") as fp:
    for i, line in enumerate(fp):
        if i == 25:
            # 26th line
        elif i == 29:
            # 30th line
        elif i > 29:
            break

反對(duì) 回復(fù) 2019-06-27

Smart貓小萌

TA貢獻(xiàn)1911條經(jīng)驗(yàn) 獲得超7個(gè)贊

一種快速和緊湊的辦法可以是：

def picklines(thefile, whatlines):
  return [x for i, x in enumerate(thefile) if i in whatlines]

這將接受任何打開(kāi)的類似文件的對(duì)象。thefile(留待調(diào)用方從磁盤(pán)文件打開(kāi)它，還是通過(guò)套接字或其他類似文件的流打開(kāi))和一組基于零的行索引。whatlines，并返回一個(gè)列表，具有較低的內(nèi)存占用和合理的速度。如果要返回的行數(shù)很大，您可能更喜歡生成器：

def yieldlines(thefile, whatlines):
  return (x for i, x in enumerate(thefile) if i in whatlines)

-注意，唯一的區(qū)別來(lái)自于使用圓角括號(hào)，而不是使用圓括號(hào)。return語(yǔ)句，分別進(jìn)行列表理解和生成器表達(dá)式。

還請(qǐng)注意，盡管提到了“行”和“文件”，多更一般的-他們會(huì)努力的任何可迭代的，無(wú)論是打開(kāi)的文件還是任何其他的，返回一個(gè)列表(或生成器)的項(xiàng)目，根據(jù)他們的累進(jìn)項(xiàng)目編號(hào)。因此，我建議使用更合適的通用名稱；-)。

反對(duì) 回復(fù) 2019-06-27