Bagaimana cara menulis data biner ke stdout di python 3?

103

Di python 2.x saya bisa melakukan ini:

import sys, array
a = array.array('B', range(100))
a.tofile(sys.stdout)

Namun sekarang, saya mendapatkan file TypeError: can't write bytes to text stream. Apakah ada pengkodean rahasia yang harus saya gunakan?

python python-3.x

— Ivan Baldin
sumber

4

Akan jauh lebih baik untuk menemukan jawaban yang akan bekerja dengan Python 2.6+ dan 3.x

— sorin

2

os.writeakan bekerja pada Py2 dan Py3.

— David Wolever

169

Cara yang lebih baik:

import sys
sys.stdout.buffer.write(b"some binary data")

— Benjamin Peterson
sumber

4

Menggunakan sys.stdout.bufferjuga memungkinkan Anda melakukan hal-hal seperti menggunakan shutil.copyfileobjbahkan ketika objek file sumber memberikan byte, dan bukan string. +1

— csl

1

Program yang menggunakan ini tidak dapat diuji di IDLE 3:AttributeError: 'PseudoOutputFile' object has no attribute 'buffer'

— Damian Yerrick

4

@DamianYerrick di IDLE (setidaknya di Windows) pythonw.exemenjalankan IDLE, yang berarti tidak ada stdout. Ini ditiru dengan tkinter. Secara fisik tidak dapat menangani byte. Dalam hal ini, .decode('UTF-8', errors='replace')string Anda, atau jalankan python3 -I <filename>untuk mendapatkan REPL daripada menggunakan IDLE.

— Artyer

Mengacaukan urutan penulisan saat menulis stderrjika menggunakan bersama print(file=sys.stderr).

— Kotauskas

15

import os
os.write(1, a.tostring())

atau, os.write(sys.stdout.fileno(), …)jika itu lebih mudah dibaca daripada 1Anda.

— Alex Martelli
sumber

Terima kasih, itu berhasil. Rasanya agak hack-ish tapi saya rasa itu bukan hal yang umum untuk dilakukan.

— Ivan Baldin

6

Masalahnya os.writeadalah Anda harus memeriksa nilai pengembalian, karena tidak menjamin bahwa semuanya akan ditulis.

— mic_e

10

Cara idiomatis untuk melakukannya, yang hanya tersedia untuk Python 3, adalah:

with os.fdopen(sys.stdout.fileno(), "wb", closefd=False) as stdout:
    stdout.write(b"my bytes object")
    stdout.flush()

Bagian yang baik adalah ia menggunakan antarmuka objek file normal, yang semua orang terbiasa dengan Python.

Perhatikan bahwa saya mengatur closefd=Falseuntuk menghindari penutupan sys.stdoutsaat keluar dari withblok. Jika tidak, program Anda tidak akan dapat mencetak ke stdout lagi. Namun, untuk jenis deskriptor file lainnya, Anda mungkin ingin melewatkan bagian itu.

— Yajo
sumber

Mengapa Anda memerah? Bukankah lebih baik membiarkan Python memutuskan kapan harus melakukan flush stdout?

— Martin

0

Jika Anda ingin menentukan pengkodean dalam python3, Anda masih dapat menggunakan perintah byte seperti di bawah ini:

import os
os.write(1,bytes('Your string to Stdout','UTF-8'))

di mana 1 adalah nomor biasa yang sesuai untuk stdout -> sys.stdout.fileno ()

Sebaliknya, jika Anda tidak peduli dengan pengkodean, gunakan saja:

import sys
sys.stdout.write("Your string to Stdout\n")

Jika Anda ingin menggunakan os.write tanpa pengkodean, coba gunakan yang di bawah ini:

import os
os.write(1,b"Your string to Stdout\n")

— Marco smdm
sumber

1

Program yang menggunakan os.write(sys.stdout.fileno(), some_bytes)tidak akan bekerja di IDLE. io.UnsupportedOperation: fileno

— Damian Yerrick

@DamianYerrick: Anda benar ... IDLE seharusnya tidak digunakan untuk menguji sesuatu seperti itu. Singkatnya: coba buka IDLE (saya punya shell python3.5.1) dan cukup impor sys dan sys.stdout.fileno () itu akan membuat Anda io error, karena dalam IDLE ini tidak didukung operasi :-) Itu selalu penting untuk mengingat di lingkungan mana Anda bekerja dan mencoba untuk mendapatkan apa yang mungkin;) Semoga ini menjelaskan pertanyaan Anda :-) Selamat berakhir pekan.

— Marco smdm

Anda hanya menyebutkan satu cara untuk menulis data biner yang sebenarnya ke stdout, yang terakhir.

— Kotauskas