Palantir打字稿顺序行和过滤器

Question

我已经创建了一个函数，允许我通过groupby和聚合函数获得一个表。由于我分组的列有很多值，我想按行降序排序（按“百分比”列，由2列划分创建），然后只保留前5行。

我想在TypeScript中实现这一点，如果可能的话，可以将其附加到groupby函数下。

谢谢你！！！

输出示例：

代码示例：

export class funzione1 {
     @Function()
        public async test1(object1: ObjectSet<object1>):
            Promise<TwoDimensionalAggregation<string>> {
                const numerators = await object1.groupBy(e => e.column1.topValues())
                                        .sum(e => e.column2);
                const denominators = await object1.groupBy(e => e.column1.topValues())
                                        .sum(e => e.column3);
        return this.divideTwoDimensional(numerators, denominators);
    }

    private divideTwoDimensional(numerators:TwoDimensionalAggregation<string>,
                             denominators: TwoDimensionalAggregation<string>):
                             TwoDimensionalAggregation<string> {

        const percentage = numerators.buckets.map((bucket, i) => {
           const numerator = bucket.value;
           const denominator = denominators.buckets[i].value;
            if (denominator === 0) {
                return { key: bucket.key, value: 0 };
            }
            return { key: bucket.key, value: numerator / denominator }
        });

        return { buckets: percentage };
    }

Answer 1

您可以使用JavaScript按百分比对桶进行排序。带有自定义比较器的sort（）函数。例如，这里的helper函数sortBucketsByValue2D使用它按桶的百分比值对桶进行排序：

import { Function, TwoDimensionalAggregation, BucketKey, BucketValue } from "@foundry/functions-api";
import { Objects, object1 } from "@foundry/ontology-api";

export class funzione1 {
    @Function()
    public async test1(object1: ObjectSet<object1>):
        Promise<TwoDimensionalAggregation<string>> {
        const numerators = await object1.groupBy(e => e.column1.topValues())
            .sum(e => e.column2);
        const denominators = await object1.groupBy(e => e.column1.topValues())
            .sum(e => e.column3);
        
        const withPercentage = divideTwoDimensional(numerators, denominators);
        return sortBucketsByValue2D(withPercentage, 'desc');
    }
}

const divideTwoDimensional = (
    numerators: TwoDimensionalAggregation<string>,
    denominators: TwoDimensionalAggregation<string>
): TwoDimensionalAggregation<string> => {
    const percentage = numerators.buckets.map((bucket, i) => {
        const numerator = bucket.value;
        const denominator = denominators.buckets[i].value;
        if (denominator === 0) {
            return { key: bucket.key, value: 0 };
        }
        return { key: bucket.key, value: numerator / denominator }
    });

    return { buckets: percentage };
}

/**
 * Sort buckets of a 2D aggregation by their value in the specified order
 * 
 * Example input 1:
 * { buckets: [
 *   { key: { min: "2022-01-01", max: "2022-12-31" }, value: 456 },
 *   { key: { min: "2021-01-01", max: "2021-12-31" }, value: 123 },
 *   { key: { min: "2023-01-01", max: "2023-12-31" }, value: 789 },
 * ]}
 * 
 * Example output 1:
 * { buckets: [
 *   { key: { min: "2021-01-01", max: "2021-12-31" }, value: 123 },
 *   { key: { min: "2022-01-01", max: "2022-12-31" }, value: 456 },
 *   { key: { min: "2023-01-01", max: "2023-12-31" }, value: 789 },
 * ]}
 * 
 * Example input 2:
 * { buckets: [
 *   { key: 17, value: 456 },
 *   { key: 21, value: 123 },
 *   { key: 23, value: 789 },
 * ]}
 * 
 * Example output 2:
 * { buckets: [
 *   { key: 21, value: 123 },
 *   { key: 17, value: 456 },
 *   { key: 23, value: 789 },
 * ]}
 */
function sortBucketsByValue2D<K extends BucketKey, V extends BucketValue>(
    buckets: TwoDimensionalAggregation<K, V>,
    order: 'asc' | 'desc' = 'asc'
): TwoDimensionalAggregation<K, V> {
    return {
        // See https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Array/sort
        buckets: buckets.buckets.sort(({ value: v1 }, { value: v2 }) => {
            // These are be either numbers, timestamps or localdates which can be compared like this
            return (order === 'desc' ? -1 : 1) * (v1.valueOf() - v2.valueOf());
        }),
    };
}

Answer 2

您可以使用百分比列表来实现这一点；它只是一个具有键/值结构的对象数组，如：

{
  key: "bucketKey",
  value: .5
}

因此，您可以像对任何其他对象列表一样对该对象列表进行排序。

类似地，排序后，可以使用本机数组切片方法返回数组的第一个元素。

如果在创建百分比数组和添加bucket对象的return语句之间执行此操作，则应该会得到所需的结果。