Manera más agradable para rellenar de ceros para una cadena
-
19-08-2019 - |
Pregunta
¿Qué es la mayoría de Python el camino a la almohadilla de una cadena numérica con ceros a la izquierda, es decir, por lo que la cadena numérica tiene una longitud específica?
Solución
Cadenas:
>>> n = '4'
>>> print(n.zfill(3))
004
Y para números:
>>> n = 4
>>> print('%03d' % n)
004
>>> print(format(n, '03')) # python >= 2.6
004
>>> print('{0:03d}'.format(n)) # python >= 2.6
004
>>> print('{foo:03d}'.format(foo=n)) # python >= 2.6
004
>>> print('{:03d}'.format(n)) # python >= 2.7 + python3
004
>>> print('{0:03d}'.format(n)) # python 3
004
>>> print(f'{n:03}') # python >= 3.6
004
Otros consejos
Simplemente use el método rjust del objeto de cadena.
Este ejemplo hará una cadena de 10 caracteres de largo, rellenando según sea necesario.
>>> t = 'test'
>>> t.rjust(10, '0')
>>> '000000test'
Además de zfill
, puede usar el formato de cadena general:
print(f'{number:05d}') # (since Python 3.6), or
print('{:05d}'.format(number)) # or
print('{0:05d}'.format(number)) # or (explicit 0th positional arg. selection)
print('{n:05d}'.format(n=number)) # or (explicit `n` keyword arg. selection)
print(format(number, '05d'))
Documentación para formato de cadena y f-strings .
>>> '99'.zfill(5)
'00099'
>>> '99'.rjust(5,'0')
'00099'
si quieres lo contrario:
>>> '99'.ljust(5,'0')
'99000'
Funciona tanto en Python 2 como en Python 3:
>>> "{:0>2}".format("1") # Works for both numbers and strings.
'01'
>>> "{:02}".format(1) # Only works for numbers.
'01'
str(n).zfill(width)
funcionará con string
s, int
s, float
s ... y es Python 2. x y 3. x compatible:
>>> n = 3
>>> str(n).zfill(5)
'00003'
>>> n = '3'
>>> str(n).zfill(5)
'00003'
>>> n = '3.0'
>>> str(n).zfill(5)
'003.0'
width = 10
x = 5
print "%0*d" % (width, x)
> 0000000005
¡Vea la documentación impresa para todos los detalles interesantes!
Actualización para Python 3.x (7.5 años después)
Esa última línea ahora debería ser:
print("%0*d" % (width, x))
I.e. print()
ahora es una función, no una declaración. Tenga en cuenta que todavía prefiero el estilo Old School printf()
porque, IMNSHO, se lee mejor, y porque, um, he estado usando esa notación desde enero de 1980. Algo ... perros viejos ... algo algo ... nuevos trucos.
Para los que vinieron aquí a entender y no solo una respuesta rápida. Hago esto especialmente para cadenas de tiempo:
hour = 4
minute = 3
"{:0>2}:{:0>2}".format(hour,minute)
# prints 04:03
"{:0>3}:{:0>5}".format(hour,minute)
# prints '004:00003'
"{:0<3}:{:0<5}".format(hour,minute)
# prints '400:30000'
"{:$<3}:{:#<5}".format(hour,minute)
# prints '4$$:3####'
" 0 " símbolos qué reemplazar con " 2 " caracteres de relleno, el valor predeterminado es un espacio vacío
" > " los símbolos alinean todos los 2 " 0 " carácter a la izquierda de la cadena
": " símbolos the format_spec
¿Qué es la mayoría de python el camino a la almohadilla de una cadena numérica con ceros a la izquierda, es decir, por lo que la cadena numérica tiene una longitud específica?
str.zfill
está diseñado específicamente para hacer esto:
>>> '1'.zfill(4)
'0001'
Tenga en cuenta que está específicamente diseñado para manejar numéricos en cadenas como lo solicita, y se mueve un +
o -
al principio de la cadena:
>>> '+1'.zfill(4)
'+001'
>>> '-1'.zfill(4)
'-001'
He aquí la ayuda en str.zfill
:
>>> help(str.zfill)
Help on method_descriptor:
zfill(...)
S.zfill(width) -> str
Pad a numeric string S with zeros on the left, to fill a field
of the specified width. The string S is never truncated.
Rendimiento
Esta es también la más eficiente de los métodos alternativos:
>>> min(timeit.repeat(lambda: '1'.zfill(4)))
0.18824880896136165
>>> min(timeit.repeat(lambda: '1'.rjust(4, '0')))
0.2104538488201797
>>> min(timeit.repeat(lambda: f'{1:04}'))
0.32585487607866526
>>> min(timeit.repeat(lambda: '{:04}'.format(1)))
0.34988890308886766
Para comparar mejor las manzanas a las manzanas de las %
método (nota es realmente más lento), que de otra forma pre-calcular:
>>> min(timeit.repeat(lambda: '1'.zfill(0 or 4)))
0.19728074967861176
>>> min(timeit.repeat(lambda: '%04d' % (0 or 1)))
0.2347015216946602
La aplicación
Con un poco de investigación, me encontré con la aplicación de la zfill
método en Objects/stringlib/transmogrify.h
:
static PyObject *
stringlib_zfill(PyObject *self, PyObject *args)
{
Py_ssize_t fill;
PyObject *s;
char *p;
Py_ssize_t width;
if (!PyArg_ParseTuple(args, "n:zfill", &width))
return NULL;
if (STRINGLIB_LEN(self) >= width) {
return return_self(self);
}
fill = width - STRINGLIB_LEN(self);
s = pad(self, fill, 0, '0');
if (s == NULL)
return NULL;
p = STRINGLIB_STR(s);
if (p[fill] == '+' || p[fill] == '-') {
/* move sign to beginning of string */
p[0] = p[fill];
p[fill] = '0';
}
return s;
}
Vamos a caminar a través de este C código.
Primero analiza el argumento de posición, lo que significa que no permiten a los parámetros de palabra clave:
>>> '1'.zfill(width=4)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: zfill() takes no keyword arguments
Entonces se comprueba si es de la misma longitud o más, en cuyo caso se devuelve la cadena.
>>> '1'.zfill(0)
'1'
zfill
llamadas pad
(este pad
la función también es llamado por ljust
, rjust
, y center
como bien).Básicamente, esto copia el contenido en una cadena nueva y llena en el relleno.
static inline PyObject *
pad(PyObject *self, Py_ssize_t left, Py_ssize_t right, char fill)
{
PyObject *u;
if (left < 0)
left = 0;
if (right < 0)
right = 0;
if (left == 0 && right == 0) {
return return_self(self);
}
u = STRINGLIB_NEW(NULL, left + STRINGLIB_LEN(self) + right);
if (u) {
if (left)
memset(STRINGLIB_STR(u), fill, left);
memcpy(STRINGLIB_STR(u) + left,
STRINGLIB_STR(self),
STRINGLIB_LEN(self));
if (right)
memset(STRINGLIB_STR(u) + left + STRINGLIB_LEN(self),
fill, right);
}
return u;
}
Después de llamar a pad
, zfill
mueve cualquier originalmente anterior +
o -
al principio de la cadena.
Tenga en cuenta que para la cadena original a ser en realidad numérico no es necesario:
>>> '+foo'.zfill(10)
'+000000foo'
>>> '-foo'.zfill(10)
'-000000foo'
Para códigos postales guardados como enteros:
>>> a = 6340
>>> b = 90210
>>> print '%05d' % a
06340
>>> print '%05d' % b
90210
Comparación rápida de tiempo:
setup = '''
from random import randint
def test_1():
num = randint(0,1000000)
return str(num).zfill(7)
def test_2():
num = randint(0,1000000)
return format(num, '07')
def test_3():
num = randint(0,1000000)
return '{0:07d}'.format(num)
def test_4():
num = randint(0,1000000)
return format(num, '07d')
def test_5():
num = randint(0,1000000)
return '{:07d}'.format(num)
def test_6():
num = randint(0,1000000)
return '{x:07d}'.format(x=num)
def test_7():
num = randint(0,1000000)
return str(num).rjust(7, '0')
'''
import timeit
print timeit.Timer("test_1()", setup=setup).repeat(3, 900000)
print timeit.Timer("test_2()", setup=setup).repeat(3, 900000)
print timeit.Timer("test_3()", setup=setup).repeat(3, 900000)
print timeit.Timer("test_4()", setup=setup).repeat(3, 900000)
print timeit.Timer("test_5()", setup=setup).repeat(3, 900000)
print timeit.Timer("test_6()", setup=setup).repeat(3, 900000)
print timeit.Timer("test_7()", setup=setup).repeat(3, 900000)
> [2.281613943830961, 2.2719342631547077, 2.261691106209631]
> [2.311480238815406, 2.318420542148333, 2.3552384305184493]
> [2.3824197456864304, 2.3457239951596485, 2.3353268829498646]
> [2.312442972404032, 2.318053102249902, 2.3054072168069872]
> [2.3482314132374853, 2.3403386400002475, 2.330108825844775]
> [2.424549090688892, 2.4346475296851438, 2.429691196530058]
> [2.3259756401716487, 2.333549212826732, 2.32049893822186]
He hecho diferentes pruebas de diferentes repeticiones. Las diferencias no son enormes, pero en todas las pruebas, la solución zfill
fue más rápida.
Cuando se usa >= Python 3.6
la forma más limpia es usar f- cadenas con formato de cadena :
>>> s = f"{1:08}" # inline
>>> s
'00000001'
>>> n = 1
>>> s = f"{n:08}" # int variable
>>> s
'00000001'
>>> c = "1"
>>> s = f"{c:0>8}" # str variable
>>> s
'00000001'
Otro enfoque sería utilizar una lista de comprensión con una condición que verifique las longitudes. A continuación se muestra una demostración:
# input list of strings that we want to prepend zeros
In [71]: list_of_str = ["101010", "10101010", "11110", "0000"]
# prepend zeros to make each string to length 8, if length of string is less than 8
In [83]: ["0"*(8-len(s)) + s if len(s) < desired_len else s for s in list_of_str]
Out[83]: ['00101010', '10101010', '00011110', '00000000']
También se podría repetir "0", anteponer a str(n)
y obtener en el extremo derecho de la anchura de corte.Rápido y sucio poca expresión.
def pad_left(n, width, pad="0"):
return ((pad * width) + str(n))[-width:]