Compare commits

..

No commits in common. "d2256b0ee949e7cb054f85386822633f3dd4a1e4" and "f8e7ed42b3b795a0bb787e3a550abe379d4cc377" have entirely different histories.

4 changed files with 286 additions and 256 deletions

File diff suppressed because one or more lines are too long

View File

@ -3,7 +3,7 @@ import unittest
class Test(unittest.TestCase):
def test_padding_array(self):
self.assertEqual(padding_array([[1,2],[3]]),[[1,2],[3,0]])
self.assertEquals(padding_array([[1,2],[3]]),[[1,2],[3,0]])
if __name__ == '__main__':
unittest.main()

View File

@ -1,35 +0,0 @@
import itertools
from typing import Tuple, TypeVar, Iterable
T = TypeVar('T')
def groupby_index(iter: Iterable[T],n:int) -> Iterable[Iterable[T]]:
"""group list by index
Args:
iter (Iterable[T]): iterator to group by index
n (int): The size of groups
Returns:
Iterable[Iterable[T]]: iterable object to group by index
>>> [*map(lambda x:[*x],groupby_index([1,2,3,4],2))]
[[1, 2], [3, 4]]
"""
def keyfunc(x: Tuple[int,T]) -> int:
k, _ = x
return (k // n)
def mapper(x: Tuple[int, Tuple[int, T]]):
_, v = x
return map(lambda y: y[1],v)
g = itertools.groupby(enumerate(iter), keyfunc)
return map(mapper,g)
if __name__ == "__main__":
test_list = [*range(20,-10,-1)]
for g in groupby_index(test_list,4):
print([*g])
print([*map(lambda x:[*x],groupby_index([1,2,3,4],2))])
for g in groupby_index([1,2,3,4],2):
print([*g])

View File

@ -1,9 +0,0 @@
import unittest
from groupby_index import *
class Test(unittest.TestCase):
def test_padding_array(self):
self.assertEqual([*map(lambda x:[*x],groupby_index([1,2,3,4],2))],[[1,2],[3,4]])
if __name__ == '__main__':
unittest.main()