From 4721943aecb8b7c7f2cf3cb726a41b5edd3e8d49 Mon Sep 17 00:00:00 2001 From: Aaron LI Date: Thu, 28 Feb 2019 21:36:37 +0800 Subject: utils/ds: Add pad_dict_list() function --- fg21sim/utils/ds.py | 30 ++++++++++++++++++++++++++---- 1 file changed, 26 insertions(+), 4 deletions(-) (limited to 'fg21sim') diff --git a/fg21sim/utils/ds.py b/fg21sim/utils/ds.py index 8b69dbc..694cbe3 100644 --- a/fg21sim/utils/ds.py +++ b/fg21sim/utils/ds.py @@ -1,5 +1,5 @@ -# Copyright (c) 2017 Weitian LI -# MIT license +# Copyright (c) 2017,2019 Weitian LI +# MIT License """ Data structure/set utilities. @@ -78,5 +78,27 @@ def dictlist_to_dataframe(dictlist, keys=None): dv = [d[k] for k in keys] dv2 = list(_flatten_list(dv)) data.append(dv2) - dataframe = pd.DataFrame(data, columns=columns) - return dataframe + + return pd.DataFrame(data, columns=columns) + + +def pad_dict_list(d, keys, length, fill=None): + """ + Pad the lists specified by ``keys`` to the given ``length``. + + Parameters + ---------- + d : dict + The dictionary to be updated. + keys : list[str] + The keys of the lists in the dictionary to be padded. + length : int + The expected length. + fill : optional + The value filled to the padded list. + """ + for k in keys: + v = d[k] + n = len(v) + if n < length: + d[k] = v + [fill] * (length - n) -- cgit v1.2.2