Як надійно розділити рядок у Python, коли він може не містити шаблон або всі n елементів?

У Perl я можу робити:

my ($x, $y) = split /:/, $str;

І це спрацює незалежно від того, чи рядок містить шаблон.

Однак у Python це не спрацює:

a, b = "foo".split(":")  # ValueError: not enough values to unpack

Який канонічний спосіб запобігти помилкам у таких випадках?

Якщо ви ділитесь лише на дві частини (як у вашому прикладі), ви можете використовувати str.partition()для отримання гарантованого аргументу розмір розпакування 3:

>>> a, sep, b = 'foo'.partition(':')
>>> a, sep, b
('foo', '', '')

str.partition() завжди повертає 3-кортеж, незалежно від того, знайдено роздільник чи ні.

Іншою альтернативою Python 3.x є використання розширеного ітеративного розпакування :

>>> a, *b = 'foo'.split(':')
>>> a, b
('foo', [])

Це призначає перший розділений елемент aі список решти елементів (якщо такі є) b.

Оскільки ви працюєте на Python 3, це легко. PEP 3132 запросив привітальне спрощення синтаксису при призначенні кортежів - Розширене ітераційне розпакування . У минулому, при присвоєнні змінним у кортежі кількість елементів зліва від завдання повинна бути точно дорівнює кількості праворуч.

У Python 3 ми можемо позначити будь-яку змінну ліворуч як список, додавши позначку зірочкою *. Це захопить якомога більше значень, одночасно заповнюючи змінні праворуч (тому це не обов’язково повинен бути самим правим елементом). Це дозволяє уникнути багатьох неприємних зрізів, коли ми не знаємо довжини кортежу.

a, *b = "foo".split(":")  
print("a:", a, "b:", b)

Дає:

a: foo b: []

РЕДАКТУЙТЕ такі коментарі та обговорення:

У порівнянні з версією Perl, це значно відрізняється, але це спосіб Python (3). У порівнянні з версією Perl, re.split()було б більше подібним, однак виклик механізму RE для розподілу навколо одного символу є непотрібними накладними витратами.

З кількома елементами в Python:

s = 'hello:world:sailor'
a, *b = s.split(":")
print("a:", a, "b:", b)

дає:

a: hello b: ['world', 'sailor']

Однак у Perl:

my $s = 'hello:world:sailor';
my ($a, $b) = split /:/, $s;
print "a: $a b: $b\n";

дає:

a: hello b: world

Видно, що додаткові елементи ігноруються або втрачаються в Perl. Це досить легко відтворити в Python, якщо потрібно:

s = 'hello:world:sailor'
a, *b = s.split(":")
b = b[0]
print("a:", a, "b:", b)

Отже, a, *b = s.split(":")еквівалент у Perl буде

my ($a, @b) = split /:/, $s;

Примітка: ми не повинні використовувати $aі $bвзагалі Perl, оскільки вони мають особливе значення при використанні sort. Я використав їх тут для узгодження з прикладом Python.

У Python є додатковий фокус в рукаві, ми можемо розпакувати будь-який елемент у кортежі зліва:

s = "one:two:three:four"
a, *b, c = s.split(':')
print("a:", a, "b:", b, "c:", c)

Дає:

a: one b: ['two', 'three'] c: four

У той час як в еквіваленті Perl, масив ( @b) жадібний, і скаляр $cє undef:

use strict;
use warnings;

my $s = 'one:two:three:four';
my ($a, @b, $c) = split /:/, $s;
print "a: $a b: @b c: $c\n";

Дає:

Use of uninitialized value $c in concatenation (.) or string at gash.pl line 8.
a: one b: two three four c: 

Ви завжди можете зловити виняток.

Наприклад:

some_string = "foo"

try:
    a, b = some_string.split(":")
except ValueError:
    a = some_string
    b = ""

Якщо присвоєння цілому оригінальному рядку aта порожньому рядку bє бажаною поведінкою, я б, мабуть, використав, str.partition()як пропонує Євгеній У. Однак це рішення дає вам більше контролю над тим, що саме відбувається, коли в рядку немає роздільника, що може бути корисним у деяких випадках.

splitзавжди буде повертати список. a, b = ...завжди очікуватиме, що довжина списку буде двома. Ви можете використовувати щось на зразок l = string.split(':'); a = l[0]; ....

Ось один вкладиш: a, b = (string.split(':') + [None]*2)[:2]

Як щодо використання регулярних виразів:

import re 
string = 'one:two:three:four'

у 3.X:

a, *b = re.split(':', string)

у 2.X:

a, b = re.split(':', string)[0], re.split(':', string)[1:]

Таким чином, ви також можете використовувати регулярні вирази для розділення (тобто \ d)