如題,刷題的時候程序需要從標(biāo)準(zhǔn)(input()
)輸入獲取數(shù)據(jù),如果每次運行代碼都需要手動的在console
復(fù)制題目的樣例輸入就很費事杯拐,輸入重定向可以解決這個問題,實行起來有兩種方式:
- 使用
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()
- 使用PyCharm的文件重定向
-
在Python腳本同級目錄下新建一個和腳本同樣名字(我的習(xí)慣)的txt文件:
- 然后在PyCharm右上角文件運行的地方找到運行配置
Edit Configurations
- 在
Execution
中勾選Rediract input from:
選項
-
勾選之后在瀏覽右邊的目錄找到條txt文件作為輸入從定向文件就行了:
-
腳本運行的時候console會顯示是從文件進行的輸入重定向: