Taking two column waterfall flow as an example, how should we build an array of each column

Ideas

Add new image elements to the column with the lowest height , Until it is longer than the other column . Then add elements to the short column

Code （ Example ）

// data Is an array containing all picture information of the waterfall stream 
// numberOfColumns  It represents the number of waterfall flow columns , Our case is 2 Column 
function classifyData(data: any[], numberOfColumns: number): any[] {
    
    const result: any[] = [];
    //  Array initialization 
    for (let i = 0; i < numberOfColumns; i++) {
    
        result.push({
    
            height: 0,
            contents: []
        });
    }
    data.forEach((item) => {
    
        const minHeight = Math.min(...result.map((list) => list.height));
        const minList = result.find((list) => list.height === minHeight);
        //  The principle of this function is to add an array after the column that is short 
        minList.contents.push(item);
        minList.height += item.width === 0 ? 0 : item.height / item.width;
    });
    return result;
}