Dictionary of Expyriment pictures from files in a subfolder

expyriment_picture_dict.py

import os as _os

def picture_dict(suffix, subfolder="", preload=False):
    """Returns a dictionary of Expyriment pictures from files in a subfolder

    Returns
    -------
    picture_dict : dictionary
        key: filename (with suffix)
        value: Expyriment stimulus

    """

    rtn = {}
    path = _os.path.dirname("." + _os.path.sep + subfolder + _os.path.sep)
    for flname in _os.listdir(path):
        if flname.endswith(suffix):
            rtn[flname] = stimuli.Picture(filename = path + _os.path.sep + flname)
            if preload:
                rtn[flname].preload()
    return rtn