使用LINQ查找C#中数字数组的累积和

时间:2011-01-28 00:17:53

标签: c# arrays linq cumulative-sum

我有一个包含双打的csv字符串(例如“0.3,0.4,0.3”),我希望能够输出一个包含这些数字累积和的双数组(例如[0.3,0.7,1.0])。

到目前为止,我有

double[] probabilities = textBox_f.Text.Split(new char[]{','}).Select(s => double.Parse(s)).ToArray();

将数字作为数组给出,但不是数字的累积和。

有没有办法继续使用这个表达式来获得我想要的东西,或者我是否需要使用迭代从我已经拥有的数组中创建一个新数组?

11 个答案:

答案 0 :(得分:45)

有一个普遍性的时间,并且有时间解决实际提出的问题。这是后者之一。如果你想创建一个方法,将一系列双精度变成一个部分和的序列,那么就这样做:

public static IEnumerable<double> CumulativeSum(this IEnumerable<double> sequence)
{
    double sum = 0;
    foreach(var item in sequence)
    {
        sum += item;
        yield return sum;
    }        
}

易。没有弄乱聚合和复杂的查询和诸如此类的东西。易于理解,易于调试,易于使用:

textBox_f.Text
    .Split(new char[]{','})
    .Select(s => double.Parse(s))
    .CumulativeSum()
    .ToArray();

现在,我注意到如果这是用户输入,那么double.Parse可以抛出异常;做一些像这样的事情可能是个更好的主意:

public static double? MyParseDouble(this string s)
{
    double d;
    if (double.TryParse(s, out d))
        return d;
    return null;
}

public static IEnumerable<double?> CumulativeSum(this IEnumerable<double?> sequence)
{
    double? sum = 0;
    foreach(var item in sequence)
    {
        sum += item;
        yield return sum;
    }        
}
...
textBox_f.Text
    .Split(new char[]{','})
    .Select(s => s.MyParseDouble())
    .CumulativeSum()
    .ToArray();

现在如果用户输入错误,你不会得到异常;你得到零。

答案 1 :(得分:19)

前段时间我有类似的要求。基本上,我需要进行聚合,但我还需要选择每个中间值。所以我编写了一个名为SelectAggregate的扩展方法(可能不是最合适的名称,但我找不到更好的名称),可以这样使用:

double[] numbers = new [] { 0.3, 0.4, 0.3 };
double[] cumulativeSums = numbers.SelectAggregate(0.0, (acc, x) => acc + x).ToArray();

以下是代码:

    public static IEnumerable<TAccumulate> SelectAggregate<TSource, TAccumulate>(
        this IEnumerable<TSource> source,
        TAccumulate seed,
        Func<TAccumulate, TSource, TAccumulate> func)
    {
        source.CheckArgumentNull("source");
        func.CheckArgumentNull("func");
        return source.SelectAggregateIterator(seed, func);
    }

    private static IEnumerable<TAccumulate> SelectAggregateIterator<TSource, TAccumulate>(
        this IEnumerable<TSource> source,
        TAccumulate seed,
        Func<TAccumulate, TSource, TAccumulate> func)
    {
        TAccumulate previous = seed;
        foreach (var item in source)
        {
            TAccumulate result = func(previous, item);
            previous = result;
            yield return result;
        }
    }

答案 2 :(得分:5)

您希望使用Aggregate运算符,并使用List<double>作为聚合累加器。这样你就可以产生一个本身就是一系列总和的投影。

这是一个让你入门的例子:

double[] runningTotal = textBox_f.Text
            .Split(new char[]{','})
            .Select(s => double.Parse(s))
            .Aggregate((IEnumerable<double>)new List<double>(), 
                       (a,i) => a.Concat(new[]{a.LastOrDefault() + i}))
            .ToArray();

答案 3 :(得分:2)

为什么需要LINQ?

var cumulative = new double[probabilities.Length];
for (int i = 0; i < probabilities.Length; i++)
    cumulative[i] = probabilities[i] + (i == 0 ? 0 : cumulative[i-1]);

答案 4 :(得分:2)

首先,我认为Linq不是一件好事。简单的旧foreach会做得更好。但作为一个谜题,它很好。

第一个想法是使用子查询,但我不喜欢它,因为它是O(n ^ 2)。这是我的线性解决方案:

        double[] probabilities = new double[] { 0.3, 0.4, 0.3};
        probabilities
            .Aggregate(
                new {sum=Enumerable.Empty<double>(), last = 0.0d},
                (a, c) => new {
                    sum = a.sum.Concat(Enumerable.Repeat(a.last+c,1)),
                    last = a.last + c
                },
                a => a.sum
            );

答案 5 :(得分:1)

使用RX:

var input=new double[]{ ... }
var output = new List<double>();
input.ToObservable().Scan((e, f) => f + e).Subscribe(output.Add);

答案 6 :(得分:0)

这是使用LINQ 的方式:

double[] doubles = { 1.7, 2.3, 1.9, 4.1, 2.9 };
var doublesSummed = new List<double>();

Enumerable.Aggregate(doubles, (runningSum, nextFactor) => {
    double currentSum = runningSum + nextFactor;
    doublesSummed.Add(currentSum);
    return currentSum;
});

doublesSummed.Dump();

在LINQPad中:

  • 4
  • 5.9
  • 10
  • 12.9

答案 7 :(得分:0)

实际上,使用generator概括起来非常简单。这是一个称为Accumulate的新扩展方法,其工作方式类似于SelectAggregate的组合。通过将二进制函数应用于序列中的每个元素以及到目前为止的累积值,它返回一个新序列。

 public static class EnumerableHelpers 
 {
    public static IEnumerable<U> Accumulate<T, U>(this IEnumerable<T> self, U init, Func<U, T, U> f) 
    {
        foreach (var x in self)
            yield return init = f(init, x);
    }

    public static IEnumerable<T> Accumulate<T>(this IEnumerable<T> self, Func<T, T, T> f)
    {
        return self.Accumulate(default(T), f);
    }

    public static IEnumerable<double> PartialSums(this IEnumerable<double> self)
    {
        return self.Accumulate((x, y) => x + y);
    }

    public static IEnumerable<int> PartialSums(this IEnumerable<int> self)
    {
        return self.Accumulate((x, y) => x + y);
    }
 }

答案 8 :(得分:0)

List<double>的累计金额:

var nums = new List<double>() { 0.3, 0.0, 0.4, 1.1 };
var cumsum = nums.Aggregate(new List<double> (), 
              (list, next) => { list.Add(list.LastOrDefault() + next); return list; });

答案 9 :(得分:0)

这是我的解决方法:

  • Linq
  • 线性时间
  • 线性记忆
  • 没有副作用

仅需注意的是,它不适用于空列表(处理起来很简单)。

    var doublesSummed  = doubles.Skip(1).Aggregate(
        new {
            sum = doubles.First(),
            doubles = new [] {doubles.First()}.AsEnumerable()
        },  
        (acc, nextDouble) => new {
            sum = acc.sum + nextDouble,
            doubles = acc.doubles.Append(acc.sum + nextDouble)
        }
    );

Demo

答案 10 :(得分:-1)

var input=new double[]{ ... }
double sum=0;

var output=input
    .Select(w=>sum+=w);