mirror of
https://github.com/erg-lang/erg.git
synced 2025-08-04 10:49:54 +00:00
26 lines
726 B
Python
26 lines
726 B
Python
from _erg_control import then__
|
|
|
|
class Array(list):
|
|
def dedup(self, same_bucket=None):
|
|
if same_bucket is None:
|
|
return Array(list(set(self)))
|
|
else:
|
|
removes = []
|
|
for lhs, rhs in zip(self, self[1:]):
|
|
if same_bucket(lhs, rhs):
|
|
removes.append(lhs)
|
|
for remove in removes:
|
|
self.remove(remove)
|
|
return self
|
|
|
|
def push(self, value):
|
|
self.append(value)
|
|
return self
|
|
|
|
def partition(self, f):
|
|
return Array(list(filter(f, self))), Array(
|
|
list(filter(lambda x: not f(x), self))
|
|
)
|
|
|
|
def __mul__(self, n):
|
|
return then__(list.__mul__(self, n), Array)
|