Python刷題的時候編寫測試用例的輸入重定向

如題,刷題的時候程序需要從標(biāo)準(zhǔn)(input())輸入獲取數(shù)據(jù),如果每次運行代碼都需要手動的在console復(fù)制題目的樣例輸入就很費事杯拐,輸入重定向可以解決這個問題,實行起來有兩種方式:

  1. 使用io.StringIO
import io
import sys


def func():
    n = int(input())
    lst_str = [input().strip() for _ in range(n)] # 多行數(shù)據(jù)
    lst_int = list(map(int, input().split())) # 一個數(shù)組
    print(lst_str)
    print(lst_int)


def main():
    sys.stdin = io.StringIO('''3
                            line 1
                            line 2
                            line 3
                            1 2 3 4 5''')
    func()


if __name__ == '__main__':
    main()

  1. 使用PyCharm的文件重定向
  • 在Python腳本同級目錄下新建一個和腳本同樣名字(我的習(xí)慣)的txt文件:



  • 然后在PyCharm右上角文件運行的地方找到運行配置Edit Configurations
    Edit Configurations
  • Execution中勾選Rediract input from:選項
    Rediract input from:
  • 勾選之后在瀏覽右邊的目錄找到條txt文件作為輸入從定向文件就行了:


  • 腳本運行的時候console會顯示是從文件進行的輸入重定向:


?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末世蔗,一起剝皮案震驚了整個濱河市端逼,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌污淋,老刑警劉巖顶滩,帶你破解...
    沈念sama閱讀 212,884評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異寸爆,居然都是意外死亡诲祸,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,755評論 3 385
  • 文/潘曉璐 我一進店門而昨,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人找田,你說我怎么就攤上這事歌憨。” “怎么了墩衙?”我有些...
    開封第一講書人閱讀 158,369評論 0 348
  • 文/不壞的土叔 我叫張陵务嫡,是天一觀的道長甲抖。 經(jīng)常有香客問我,道長心铃,這世上最難降的妖魔是什么准谚? 我笑而不...
    開封第一講書人閱讀 56,799評論 1 285
  • 正文 為了忘掉前任,我火速辦了婚禮去扣,結(jié)果婚禮上柱衔,老公的妹妹穿的比我還像新娘。我一直安慰自己愉棱,他們只是感情好唆铐,可當(dāng)我...
    茶點故事閱讀 65,910評論 6 386
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著奔滑,像睡著了一般艾岂。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上朋其,一...
    開封第一講書人閱讀 50,096評論 1 291
  • 那天王浴,我揣著相機與錄音,去河邊找鬼梅猿。 笑死氓辣,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的粒没。 我是一名探鬼主播筛婉,決...
    沈念sama閱讀 39,159評論 3 411
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼癞松!你這毒婦竟也來了爽撒?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 37,917評論 0 268
  • 序言:老撾萬榮一對情侶失蹤响蓉,失蹤者是張志新(化名)和其女友劉穎硕勿,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體枫甲,經(jīng)...
    沈念sama閱讀 44,360評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡源武,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,673評論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了想幻。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片粱栖。...
    茶點故事閱讀 38,814評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖脏毯,靈堂內(nèi)的尸體忽然破棺而出闹究,到底是詐尸還是另有隱情,我是刑警寧澤食店,帶...
    沈念sama閱讀 34,509評論 4 334
  • 正文 年R本政府宣布渣淤,位于F島的核電站赏寇,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏价认。R本人自食惡果不足惜嗅定,卻給世界環(huán)境...
    茶點故事閱讀 40,156評論 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望用踩。 院中可真熱鬧渠退,春花似錦、人聲如沸捶箱。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,882評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽丁屎。三九已至荠锭,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間晨川,已是汗流浹背证九。 一陣腳步聲響...
    開封第一講書人閱讀 32,123評論 1 267
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留共虑,地道東北人愧怜。 一個月前我還...
    沈念sama閱讀 46,641評論 2 362
  • 正文 我出身青樓,卻偏偏與公主長得像妈拌,于是被迫代替她去往敵國和親拥坛。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 43,728評論 2 351

推薦閱讀更多精彩內(nèi)容